Fix function plot_repartition

@@ -45,10 +45,12 @@ def import_ncRNA(path):
 '''
 '''
 Plot function
 Plot function
 '''
 '''
-def plot_repartition(data,m,n,name,labels_names):
+def plot_repartition(data,m,n,name):
 	repartition = data[1]
 	repartition = data[1]
+	labels_names = data[0]
 	dico = []
 	dico = []
 	N = np.sum(repartition)
 	N = np.sum(repartition)
+	print(labels_names)
 	for i in range(m):
 	for i in range(m):
 		value_classes = np.sum(repartition,axis=0)
 		value_classes = np.sum(repartition,axis=0)
 		for j in range(n):
 		for j in range(n):

@@ -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")
 	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)
 	rep,_ = som.repartition_map(data,y)
-	plot_repartition(rep,som.dim[0],som.dim[1],path+"plot_repartition",y_label)
+	plot_repartition(rep,som.dim[0],som.dim[1],path+"plot_repartition")
 	plot_density(np.array(y_label), proba,path+"plot_density.png")
 	plot_density(np.array(y_label), proba,path+"plot_density.png")
 
 
 def main():
 def main():