setting maxiterations to 10

dice-group · Oct 4, 2018 · 6b621e8 · 6b621e8
1 parent d491eb6
commit 6b621e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/org/aksw/word2vecrestful/word2vec/W2VNrmlMemModelKMeans.java b/src/main/java/org/aksw/word2vecrestful/word2vec/W2VNrmlMemModelKMeans.java
@@ -49,7 +49,7 @@ public W2VNrmlMemModelKMeans(final Map<String, float[]> word2vec, final int vect
 	}
 
 	private void generateComparisonVectors() {
-		KMeansPlusPlusClusterer<ClusterableVec> clusterer = new KMeansPlusPlusClusterer<>(compareVecCount);
+		KMeansPlusPlusClusterer<ClusterableVec> clusterer = new KMeansPlusPlusClusterer<>(compareVecCount, 10);
 		List<ClusterableVec> vecList = new ArrayList<>();
 		for (float[] vec : word2vec.values()) {
 			vecList.add(getClusterablePoint(vec));