Esempio n. 1
0
C_FeatLabels = ['fathmm-MKL_coding_pred','MetaLR_pred','MetaSVM_pred']
D_Labels = ['Polyphen2_HDIV_pred','MutationTaster_pred','FATHMM_pred'] 
D_FeatLabels = ['Polyphen2_HDIV_pred','MutationTaster_pred','FATHMM_pred'] 

C_Positive = GetPositiveFeatureVector('training.xls',C_Labels,12)
D_Positive = GetPositiveFeatureVector('training.xls',D_Labels,12)
C_Positive_t = GetPositiveFeatureVector('testing.xls',C_Labels,4)
D_Positive_t = GetPositiveFeatureVector('testing.xls',D_Labels,4)
C_Negative,C_TestVec = GetNegativeFeatureVector('10128.annotation.xls',C_Labels)
D_Negative,D_TestVec = GetNegativeFeatureVector('10128.annotation.xls',D_Labels)
#############################################################################
print len(D_Positive)
C_dataset = C_Negative + C_Positive
D_dataset = D_Negative + D_Positive
#print dataset
C_Tree = DecisionTree.CreatTree(C_dataset,C_Labels)
D_Tree = DecisionTree.CreatTree(D_dataset,D_Labels)

#TreePlotter.CreatPlot(D_Tree)
#TreePlotter.CreatPlot(C_Tree)


C_TreeFile = open('C_Tree','w')
D_TreeFile = open('D_Tree','w')
pickle.dump(C_Tree,C_TreeFile)
pickle.dump(D_Tree,D_TreeFile)
C_TreeFile.close();D_TreeFile.close()


ClassList = [] 
#print Positive[50:]  # [50:]: #Positive:       #TestVec: