k-means - added verbose-level print after initialization

jaberg · GaelVaroquaux · commit 0e71824ca9a9 · 2011-02-24T21:49:14.000+01:00
diff --git a/scikits/learn/cluster/k_means_.py b/scikits/learn/cluster/k_means_.py
@@ -163,6 +163,8 @@ def k_means(X, k, init='k-means++', n_init=10, max_iter=300, verbose=0,
                 "be 'k-mean++' or 'random' or an ndarray, "
                 "'%s' (type '%s') was passed.")
 
+        if verbose:
+            print 'Initialization complete' 
         # iterations
         for i in range(max_iter):
             centers_old = centers.copy()

Original file line number	Diff line number	Diff line change
`@@ -163,6 +163,8 @@ def k_means(X, k, init='k-means++', n_init=10, max_iter=300, verbose=0,`
`163`	`163`	`"be 'k-mean++' or 'random' or an ndarray, "`
`164`	`164`	`"'%s' (type '%s') was passed.")`
`165`	`165`
	`166`	`+ if verbose:`
	`167`	`+ print 'Initialization complete'`
`166`	`168`	`# iterations`
`167`	`169`	`for i in range(max_iter):`
`168`	`170`	`centers_old = centers.copy()`