From 9aec207cb0722ed298e1cfd8acf39f89d177f27b Mon Sep 17 00:00:00 2001 From: Mathias Quillot Date: Mon, 22 Jul 2019 12:06:03 +0200 Subject: [PATCH] little change --- bin/cluster_kmeans.py | 1 + 1 file changed, 1 insertion(+) diff --git a/bin/cluster_kmeans.py b/bin/cluster_kmeans.py index 230757b..68b110a 100644 --- a/bin/cluster_kmeans.py +++ b/bin/cluster_kmeans.py @@ -42,6 +42,7 @@ X = np.asarray([feat_ind[x[0][0]][x[0][3]][1] for x in lst]) 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) -- 1.8.2.3