Fix rejection

@@ -61,8 +61,11 @@ To keep this files, use the parameter --keep\_features.
 
 
 Default usage:
 Default usage:
 ``` bash
 ``` bash
-python scripts/predict.py --featurer=${path_IRSOM}/bin/Featurer --file=fasta_file.fasta --model=${path_IRSOM}/model/species/ --output=output_dir_of_result
+python scripts/predict.py --featurer=${path_IRSOM}/bin/Featurer --file=fasta_file.fasta --model=${path_IRSOM}/model/species/ --output=output_dir_of_result [--reject=${rejection_threshold}]
 ```
 ```
 
 
+The rejection threshold can be set with the option `--reject`.
+By default there is no rejection.
+
 As for the train script, the features are removed by default.
 As for the train script, the features are removed by default.
 To keep them, use the parameter --keep\_features.
 To keep them, use the parameter --keep\_features.

@@ -28,7 +28,7 @@ from SLSOM.SOM import *
 from SLSOM.util import *
 from SLSOM.util import *
 
 
 def save_pred(som,data,data_names,y,proba,bmu,path):
 def save_pred(som,data,data_names,y,proba,bmu,path):
-	y_label = ["Noncoding" if x==1 else if x==0 "Coding" else "Rejected" for x in y]
+	y_label = ["Noncoding" if x==1 else "Coding" if x==0 else "Rejected" for x in y]
 	res = np.array([
 	res = np.array([
 		[data_names[i],bmu[i],y_label[i]]+[proba[i,j] for j in range(proba.shape[1])]
 		[data_names[i],bmu[i],y_label[i]]+[proba[i,j] for j in range(proba.shape[1])]
 		for i in range(data.shape[0])])
 		for i in range(data.shape[0])])