config change

dice-group · Oct 8, 2018 · fcf91e4 · fcf91e4
1 parent ef6aa6e
commit fcf91e4
Showing 1 changed file with 3 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,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<String, float[]> word2vec, final int vect
 	}
 
 	private void generateComparisonVectors() {
-		KMeansPlusPlusClusterer<ClusterableVec> clusterer = new KMeansPlusPlusClusterer<>(compareVecCount, 10);
+		KMeansPlusPlusClusterer<ClusterableVec> clusterer = new KMeansPlusPlusClusterer<>(compareVecCount, kMeansMaxItr);
 		List<ClusterableVec> vecList = new ArrayList<>();
 		for (float[] vec : word2vec.values()) {
 			vecList.add(getClusterablePoint(vec));