Ludovic PLATON

Fixing plot method for rejected transcript

......@@ -45,8 +45,7 @@ def import_ncRNA(path):
'''
Plot function
'''
def plot_repartition(data,m,n,name):
labels_names = [str(x) for x in data[0]]
def plot_repartition(data,m,n,name,labels_names):
repartition = data[1]
dico = []
N = np.sum(repartition)
......
......@@ -37,7 +37,7 @@ def save_pred(som,data,data_names,y,proba,bmu,path):
np.savetxt(path+"result.txt",res,header="Sequence_name\tBMU\tlabel"+"".join(["\tproba_class_"+str(i) for i in range(proba.shape[1])]),delimiter="\t",fmt="%s")
rep,_ = som.repartition_map(data,y)
plot_repartition(rep,som.dim[0],som.dim[1],path+"plot_repartition")
plot_repartition(rep,som.dim[0],som.dim[1],path+"plot_repartition",y_label)
plot_density(np.array(y_label), proba,path+"plot_density.png")
def main():
......