k-means - added "verbose" parameter to KMeans class

jaberg · GaelVaroquaux · commit 590f8379bf26 · 2011-02-24T21:50:15.000+01:00
diff --git a/scikits/learn/cluster/k_means_.py b/scikits/learn/cluster/k_means_.py
@@ -371,18 +371,20 @@ class KMeans(BaseEstimator):
     """
 
 
-    def __init__(self, k=8, init='random', n_init=10, max_iter=300):
+    def __init__(self, k=8, init='random', n_init=10, max_iter=300,
+            verbose=0):
         self.k = k
         self.init = init
         self.max_iter = max_iter
         self.n_init = n_init
+        self.verbose = verbose
 
     def fit(self, X, **params):
         """ Compute k-means"""
         X = np.asanyarray(X)
         self._set_params(**params)
         self.cluster_centers_, self.labels_, self.inertia_ = k_means(X,
                     k=self.k, init=self.init, n_init=self.n_init,
-                    max_iter=self.max_iter)
+                    max_iter=self.max_iter, verbose=self.verbose)
         return self
 
