Merge pull request #15 from MinimaxHeap/master

fixed bug: _k_means._centers_dense Index out of bounds
ndanielsen · Apr 27, 2021 · 4ca53d1 · 4ca53d1
2 parents 50cbca3 + d4dcdbc
commit 4ca53d1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/clustering/equal_groups.py b/clustering/equal_groups.py
@@ -495,7 +495,7 @@ def _kmeans_single(X, n_clusters, x_squared_norms, max_iter=300,
                             precompute_distances=precompute_distances,
                             distances=distances)
 
-        sample_weight = [1.0] * np.asarray(len(labels))
+        sample_weight = np.asarray([1.0] * len(labels))
         # computation of the means is also called the M-step of EM
         if sp.issparse(X):
             centers = _k_means._centers_sparse(X, sample_weight, labels, n_clusters,