little change

Mathias Quillot
1 parent 60d1f63cd5
Showing 1 changed file with 1 additions and 0 deletions Side-by-side Diff
bin/cluster_kmeans.py
@@ -42,6 +42,7 @@
 Ks = range(KMIN, KMAX+1)
 for k in Ks:
     kmeans = KMeans(n_clusters=k, n_init=10, random_state=0).fit(X)
+    preds = kmeans.predict(X)
     if ALLINDIR is False:
         subdir = str(k)
         dirname=path.join(OUTDIR, subdir)
...	...	@@ -42,6 +42,7 @@
42	42	Ks = range(KMIN, KMAX+1)
43	43	for k in Ks:
44	44	kmeans = KMeans(n_clusters=k, n_init=10, random_state=0).fit(X)
	45	+ preds = kmeans.predict(X)
45	46	if ALLINDIR is False:
46	47	subdir = str(k)
47	48	dirname=path.join(OUTDIR, subdir)