From 80a41c06c1ae44e21f870a121e7e16dc5c50036c Mon Sep 17 00:00:00 2001 From: Dylan Stewart <94502285+drstewart19@users.noreply.github.com> Date: Wed, 13 Apr 2022 12:40:10 -0400 Subject: [PATCH] Update KMeans class information Update description of k-means|| description --- dask_ml/cluster/k_means.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/dask_ml/cluster/k_means.py b/dask_ml/cluster/k_means.py index 4b02c2fa4..b1dcdd905 100644 --- a/dask_ml/cluster/k_means.py +++ b/dask_ml/cluster/k_means.py @@ -39,7 +39,8 @@ class KMeans(TransformerMixin, BaseEstimator): init : {'k-means||', 'k-means++' or ndarray} Method for center initialization, defaults to 'k-means||'. - 'k-means||' : selects the the gg + 'k-means||' : selects the parallel initalization procedure + for initializing the cluster centers. 'k-means++' : selects the initial cluster centers in a smart way to speed up convergence. Uses scikit-learn's implementation.