Méthode computePredictionFile corrigée avec les +1

TP.py

@@ -5,7 +5,6 @@ import pandas as pd
 import numpy as np
 from sklearn.metrics import accuracy_score
 from sklearn.model_selection import train_test_split
-from sklearn.naive_bayes import GaussianNB
 
 
 def buildSampleFromPath(path1, path2, size=0):
@@ -151,7 +150,11 @@ def computePredictionFile(classifier):
 
     for image in S:
         image_name = os.path.split(image["name_path"])[1]
-        file.write(image_name+" "+str(image["y_predicted_class"])+"\n")
+        if image["y_predicted_class"] == 1:
+            image_predicted_class = "+1"
+        else:
+            image_predicted_class = "-1"
+        file.write(image_name+" "+image_predicted_class+"\n")
 
     file.write("# EE = "+ str(computeError(S_train))+"\n")
     file.write("# ER = "+ str(computeError(S_test))+"\n")