Skip to content

Commit 0895f9d

Browse files
committed
"The check that init has a valid type and value is already taken care of by a generic validation tool." @jeremiedbb
1 parent 1f42d0c commit 0895f9d

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

sklearn/cluster/_kmeans.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -906,13 +906,8 @@ def _check_params_vs_input(self, X, default_n_init=None):
906906
self._n_init = default_n_init
907907
elif callable(self.init):
908908
self._n_init = default_n_init
909-
elif _is_arraylike_not_scalar(self.init):
909+
else: # array-like
910910
self._n_init = 1
911-
else:
912-
raise ValueError(
913-
'Expect init to be one of ["k-means++", "random", callable or'
914-
"array-like of shape(n_clusters, n_features)]"
915-
)
916911

917912
if _is_arraylike_not_scalar(self.init) and self._n_init != 1:
918913
warnings.warn(

0 commit comments

Comments
 (0)