From 62137dde35fdc9539f35ff5a5ee138a813d4fa34 Mon Sep 17 00:00:00 2001 From: "nikit91@gmail.com" Date: Fri, 19 Oct 2018 08:03:34 +0200 Subject: [PATCH] changing test config --- .../word2vecrestful/NrmlzdThetaMdlPrfmncTester.java | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/src/test/java/org/aksw/word2vecrestful/NrmlzdThetaMdlPrfmncTester.java b/src/test/java/org/aksw/word2vecrestful/NrmlzdThetaMdlPrfmncTester.java index afb52b3..e1245d3 100644 --- a/src/test/java/org/aksw/word2vecrestful/NrmlzdThetaMdlPrfmncTester.java +++ b/src/test/java/org/aksw/word2vecrestful/NrmlzdThetaMdlPrfmncTester.java @@ -6,7 +6,7 @@ import org.aksw.word2vecrestful.utils.Cfg; import org.aksw.word2vecrestful.word2vec.W2VNrmlMemModelBruteForce; -import org.aksw.word2vecrestful.word2vec.W2VNrmlMemModelKMeans; +import org.aksw.word2vecrestful.word2vec.W2VNrmlMemModelBinSrch; import org.aksw.word2vecrestful.word2vec.Word2VecFactory; import org.aksw.word2vecrestful.word2vec.Word2VecModel; import org.apache.log4j.LogManager; @@ -38,12 +38,12 @@ public void testNbmTime() throws IOException { LOG.info("Starting BruteForce-Model Test"); List correctWords = getCorrectWords(centroids, nbm); LOG.info("Correct Words are :" + correctWords); - LOG.info("Initializing W2VNrmlMemModelKMeans Model"); - final W2VNrmlMemModelKMeans memModel = new W2VNrmlMemModelKMeans(nbm.word2vec, nbm.vectorSize); + LOG.info("Initializing W2VNrmlMemModelBinSrch Model"); + final W2VNrmlMemModelBinSrch memModel = new W2VNrmlMemModelBinSrch(nbm.word2vec, nbm.vectorSize); memModel.process(); List lrModelWords = new ArrayList<>(); - LOG.info("Starting W2VNrmlMemModelKMeans Test"); + LOG.info("Starting W2VNrmlMemModelBinSrch Test"); for (int i = 0; i < centroids.length; i++) { LOG.info("Sending query for Centroid " + (i + 1)); @@ -53,7 +53,7 @@ public void testNbmTime() throws IOException { totTime += diff; LOG.info("Query time recorded for Centroid " + (i + 1) + " is " + diff + " milliseconds."); } - LOG.info("Average query time for W2VNrmlMemModelKMeans is : " + (totTime / centroids.length) + " milliseconds"); + LOG.info("Average query time for W2VNrmlMemModelBinSrch is : " + (totTime / centroids.length) + " milliseconds"); LOG.info("Predicted Words are :" + lrModelWords); float percVal = calcPercScore(correctWords, lrModelWords); LOG.info("Score for Test is : " + percVal + "%");