Skip to content

Commit f4e1658

Browse files
Added explanatory comment
1 parent ddc9852 commit f4e1658

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

sklearn/cluster/k_means_.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -454,6 +454,8 @@ def _kmeans_single(X, n_clusters, x_squared_norms, max_iter=300,
454454
break
455455

456456
if shift > 0:
457+
# rerun E-step in case of non-convergence so that predicted labels
458+
# match cluster centers
457459
best_labels, best_inertia = \
458460
_labels_inertia(X, x_squared_norms, best_centers,
459461
precompute_distances=precompute_distances,

0 commit comments

Comments
 (0)