Skip to content

Commit f41c98c

Browse files
committed
ENH fix int type in kmeans
1 parent 262cbdf commit f41c98c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

sklearn/cluster/k_means_.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -421,7 +421,7 @@ def _labels_inertia_precompute_dense(X, x_squared_norms, centers):
421421
k = centers.shape[0]
422422
distances = euclidean_distances(centers, X, x_squared_norms,
423423
squared=True)
424-
labels = np.empty(n_samples, dtype=np.int)
424+
labels = np.empty(n_samples, dtype=np.int32)
425425
labels.fill(-1)
426426
mindist = np.empty(n_samples)
427427
mindist.fill(np.infty)

0 commit comments

Comments
 (0)