add name to training accuracy graph

hotosm · Nov 11, 2024 · 4a1f2a6 · 4a1f2a6
1 parent 0a5f7c0
commit 4a1f2a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hot_fair_utilities/training/ramp/run_training.py b/hot_fair_utilities/training/ramp/run_training.py
@@ -353,7 +353,7 @@ def run_main_train_code(cfg):
 
     plt.legend()
     plt.savefig(
-        f"{cfg['graph_location']}/training_validation_sparse_categorical_accuracy.png"
+        f"{cfg['graph_location']}/training_accuracy.png"
     )
     plt.clf()
     print(f"Graph generated at : {cfg['graph_location']}")