diff --git a/src/main/java/org/aksw/word2vecrestful/word2vec/W2VNrmlMemModelKMeans.java b/src/main/java/org/aksw/word2vecrestful/word2vec/W2VNrmlMemModelKMeans.java index 2326182..fc4086c 100644 --- a/src/main/java/org/aksw/word2vecrestful/word2vec/W2VNrmlMemModelKMeans.java +++ b/src/main/java/org/aksw/word2vecrestful/word2vec/W2VNrmlMemModelKMeans.java @@ -30,9 +30,9 @@ public class W2VNrmlMemModelKMeans implements GenWord2VecModel { private float[][] comparisonVecs = null; private String[] wordArr; private float[][] vecArr; - private int compareVecCount = 10; + private int compareVecCount = 150; private int bucketCount = 10; - private int kMeansMaxItr = 10; + private int kMeansMaxItr = 5; private BitSet[][] csBucketContainer; // TODO : Remove this private TimeLogger tl = new TimeLogger();