Fixed issues, k was equal to 5 instead of the k parameter

quillotm
1 parent 0774ae544e
Showing 1 changed file with 1 additions and 1 deletions Side-by-side Diff
volia/clustering_modules/kmeans_multidistance.py
@@ -66,7 +66,7 @@
         # Learning k-means model
         results = []
         for i in range(ninit):
-            model = KMeans(k=5,
+            model = KMeans(k=k,
                            maxiter=maxiter,
                            distance=self.distance,
                            record_heterogeneity=[],
...	...	@@ -66,7 +66,7 @@
66	66	# Learning k-means model
67	67	results = []
68	68	for i in range(ninit):
69		- model = KMeans(k=5,
	69	+ model = KMeans(k=k,
70	70	maxiter=maxiter,
71	71	distance=self.distance,
72	72	record_heterogeneity=[],