diff --git a/src/main/java/org/aksw/word2vecrestful/word2vec/W2VNrmlMemModelKMeans.java b/src/main/java/org/aksw/word2vecrestful/word2vec/W2VNrmlMemModelKMeans.java index 944ebf0..2326182 100644 --- a/src/main/java/org/aksw/word2vecrestful/word2vec/W2VNrmlMemModelKMeans.java +++ b/src/main/java/org/aksw/word2vecrestful/word2vec/W2VNrmlMemModelKMeans.java @@ -30,8 +30,9 @@ public class W2VNrmlMemModelKMeans implements GenWord2VecModel { private float[][] comparisonVecs = null; private String[] wordArr; private float[][] vecArr; - private int compareVecCount = 100; + private int compareVecCount = 10; private int bucketCount = 10; + private int kMeansMaxItr = 10; private BitSet[][] csBucketContainer; // TODO : Remove this private TimeLogger tl = new TimeLogger(); @@ -51,7 +52,7 @@ public W2VNrmlMemModelKMeans(final Map word2vec, final int vect } private void generateComparisonVectors() { - KMeansPlusPlusClusterer clusterer = new KMeansPlusPlusClusterer<>(compareVecCount, 10); + KMeansPlusPlusClusterer clusterer = new KMeansPlusPlusClusterer<>(compareVecCount, kMeansMaxItr); List vecList = new ArrayList<>(); for (float[] vec : word2vec.values()) { vecList.add(getClusterablePoint(vec));