Bug fix in MeanShift.predict. Allowing sparse data now.

Ana Milanova · Ana Milanova · commit 7b8803649556 · 2021-08-20T20:20:44.000-04:00
diff --git a/sklearn/cluster/_mean_shift.py b/sklearn/cluster/_mean_shift.py
@@ -496,6 +496,6 @@ def predict(self, X):
             Index of the cluster each sample belongs to.
         """
         check_is_fitted(self)
-        X = self._validate_data(X, reset=False)
+        X = self._validate_data(X, accept_sparse='csr', reset=False)
         with config_context(assume_finite=True):
             return pairwise_distances_argmin(X, self.cluster_centers_)
