Keras : classificateur spirales

02-intro_rna/06-keras-tf_playground-spiral.py

@@ -173,13 +173,15 @@ os.remove("model.png") # Supression du fichier temporaire
 
 # Courbes d'apprentissage
 apts_ax.set_title("Courbes d'apprentissage")
-apts_ax.plot(apts.epoch, apts.history['loss'], 'b-', label="Perte - entrainement")
-apts_ax.plot(apts.epoch, apts.history['val_loss'], 'r-', label="Perte - validation")
-apts_ax.plot(apts.epoch, apts.history['accuracy'], 'c-', label="Précision - entrainement")
-apts_ax.plot(apts.epoch, apts.history['val_accuracy'], 'm-', label="Précision - validation")
+apts_ax.plot(apts.epoch, apts.history['loss'], 'b-', label="Entrainement")
+apts_ax.plot(apts.epoch, apts.history['val_loss'], 'r-', label="Validation")
+apts_ax.plot(apts.epoch, apts.history['accuracy'], '-', color="xkcd:light blue")
+apts_ax.plot(apts.epoch, apts.history['val_accuracy'], '-', color="xkcd:light red")
 apts_ax.set(ylim=(-0.05, 1.05))
+apts_ax.set_ylabel("Perte et précision (couleurs claires)")
 apts_ax.set_xlabel("Époque")
-apts_ax.legend()
+apts_ax.legend(loc="upper left")
+# apts_ax.legend()
 
 # Plot des données
 donnees_ax.set_title("Données")