comparison evaluation.py @ 2:b8e32e577597 draft

planemo upload for repository https://github.com/eteriSokhoyan/galaxytools/tree/branchForIterations/tools/GraphClust/CollectResults commit 9bc3c9b613d106098a78e16534897c88a3738c07
author rnateam
date Sat, 24 Dec 2016 18:08:36 -0500
parents ed8c7191b322
children 79b9117aef01
comparison
equal deleted inserted replaced
1:ed8c7191b322 2:b8e32e577597
1 import glob 1 import glob
2 from os import system 2 from os import system
3 import re 3 import re
4 from sklearn import metrics
4 5
5 def sh(script): 6 def sh(script):
6 system("bash -c '%s'" % script) 7 system("bash -c '%s'" % script)
7 8
8 dataNames = "FASTA/data.names" 9 dataNames = "FASTA/data.names"
44 toWrite = "" 45 toWrite = ""
45 for i in range(len(listOfClusters)): 46 for i in range(len(listOfClusters)):
46 toWrite += listOfClasses[i] + "\t" + listOfClusters[i] + '\n' 47 toWrite += listOfClasses[i] + "\t" + listOfClusters[i] + '\n'
47 with open("RESULTS/fullTab.tabular", "w") as full: 48 with open("RESULTS/fullTab.tabular", "w") as full:
48 full.write(toWrite) 49 full.write(toWrite)
50
51
52 listOfClasses = []
53 listOfClusters = []
54 pattern = re.compile("^RF.*$")
55
56
57 if len(listOfClasses) > 0 and pattern.match(str(listOfClasses[0])):
58 with open("RESULTS/fullTab.tabular", "r") as tabF:
59 for line in tabF.readlines():
60 listOfClasses.append(line.split()[0])
61 listOfClusters.append(line.split()[1])
62
63 completeness_score = metrics.completeness_score(listOfClasses, listOfClusters)
64 homogeneity_score = metrics.homogeneity_score(listOfClasses, listOfClusters)
65 adjusted_rand_score = metrics.adjusted_rand_score(listOfClasses, listOfClusters)
66 adjusted_mutual_info_score = metrics.adjusted_mutual_info_score(listOfClasses, listOfClusters)
67 v_measure_score = metrics.v_measure_score(listOfClasses, listOfClusters)
68
69 toWrite = "completeness_score : " + str(completeness_score) + "\n" + "homogeneity_score : " + str(homogeneity_score) + "\n" + "adjusted_rand_score : " +str(adjusted_rand_score) + "\n" + "adjusted_mutual_info_score : " + str(adjusted_mutual_info_score)+ "\n" + "v_measure_score : " + str(v_measure_score)
70
71 else:
72 toWrite = "completeness_score : NA \nhomogeneity_score : NA \nadjusted_rand_score : NA \nadjusted_mutual_info_score : NA \nv_measure_score : NA"
73
74 with open("RESULTS/evaluation.txt", "w") as fOut:
75 fOut.write(toWrite)