Adapted plot labels to new model testing.

ANN_Training.py

@@ -14,7 +14,7 @@ TODO: Add log to pipeline
 TODO: Remove object set-up
 TODO: Add testing rule to Snakefile -> Done
 TODO: Give option to change 'num_epochs' for testing -> Done
-TODO: Adapt plot labels to new set-up
+TODO: Adapt plot labels to new set-up -> Done
 
 """
 import numpy as np
@@ -126,7 +126,7 @@ class ModelTrainer(object):
         # print(np.array(classification_stats).mean(axis=0))
         classification_stats = np.array(classification_stats).mean(axis=0)
 
-        plot_classification_accuracy(*classification_stats, ['real'])
+        plot_classification_accuracy([self._model_name], *classification_stats)
 
         # Set paths for plot files if not existing already
         plot_dir = self._dir + '/model evaluation'

Plotting.py

@@ -235,7 +235,7 @@ def calculate_exact_solution(mesh, cell_len, wave_speed, final_time, interval_le
     return grid, exact
 
 
-def plot_classification_accuracy(precision, recall, accuracy, xlabels):
+def plot_classification_accuracy(xlabels, precision, recall, accuracy):
     """Plots classification accuracy.
 
     Plots the accuracy, precision, and recall in a bar plot.