fixed bug: _k_means._centers_dense Index out of bounds

ndanielsen · May 6, 2020 · d4dcdbc · d4dcdbc
1 parent 50cbca3
commit d4dcdbc
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,