We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 78ef2f2 commit 832dfc1Copy full SHA for 832dfc1
sklearn_extra/robust/_robust_weighted_estimator_helper.pyx
@@ -67,7 +67,7 @@ cpdef np.ndarray[floating] _kmeans_loss(np.ndarray[floating, ndim=2, mode='c'] X
67
np.ndarray[floating, ndim=2] centers = np.zeros([n_classes,
68
n_features],
69
dtype = dtype)
70
- np.ndarray[int32_t] num_in_cluster = np.zeros(n_classes, dtype = int)
+ np.ndarray[floating] num_in_cluster = np.zeros(n_classes, dtype = dtype)
71
np.ndarray[floating] inertias = np.zeros(n_samples, dtype = dtype)
72
for i in range(n_samples):
73
for j in range(n_features):
0 commit comments