From 3c507ecf9235b671db96359907e3246806278031 Mon Sep 17 00:00:00 2001 From: ditenberg Date: Sat, 29 Oct 2016 09:49:12 -0400 Subject: [PATCH] Change fit() docstrings In reference to #7772 --- sklearn/cluster/k_means_.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/sklearn/cluster/k_means_.py b/sklearn/cluster/k_means_.py index e91e38cb9dd31..778f1494371cc 100644 --- a/sklearn/cluster/k_means_.py +++ b/sklearn/cluster/k_means_.py @@ -874,6 +874,7 @@ def fit(self, X, y=None): Parameters ---------- X : array-like or sparse matrix, shape=(n_samples, n_features) + Training instances to cluster. """ random_state = check_random_state(self.random_state) X = self._check_fit_data(X) @@ -1308,8 +1309,8 @@ def fit(self, X, y=None): Parameters ---------- - X : array-like, shape = [n_samples, n_features] - Coordinates of the data points to cluster + X : array-like or sparse matrix, shape=(n_samples, n_features) + Training instances to cluster. """ random_state = check_random_state(self.random_state) X = check_array(X, accept_sparse="csr", order='C',