Mercurial > repos > bimib > marea

diff Marea/marea_cluster.py @ 7:771a40335876 draft

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
enchanted .py
author bimib
date Thu, 15 Nov 2018 06:12:15 -0500
parents 5721182715a7
children 7c76e8e319c2
line wrap: on
line diff
--- a/Marea/marea_cluster.py	Thu Nov 15 05:32:33 2018 -0500
+++ b/Marea/marea_cluster.py	Thu Nov 15 06:12:15 2018 -0500
@@ -554,7 +554,7 @@
         distorsion.append(tmp_kmeans.inertia_)
         predict = tmp_kmeans.predict(cluster_data)
         predict = [x+1 for x in predict]
-        classe = (pd.DataFrame(zip(cluster_data.index, predict))).astype(str)
+        classe = (pd.DataFrame(list(zip(cluster_data.index, predict)))).astype(str)
         dest = 'cluster_out/K=' + str(i) + '_' + args.name+'.tsv'
         classe.to_csv(dest, sep = '\t', index = False,
                       header = ['Patient_ID', 'Class'])