config change

dice-group · Oct 8, 2018 · f50d36b · f50d36b
1 parent fcf91e4
commit f50d36b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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();