From c5565d64fd6232ef4ba1b87426aea38b0c9b8dd1 Mon Sep 17 00:00:00 2001 From: AdamOsiowy123 Date: Wed, 4 May 2022 20:41:25 +0200 Subject: [PATCH] fix --- Scripts/evaluate_neural_network.py | 1 + 1 file changed, 1 insertion(+) diff --git a/Scripts/evaluate_neural_network.py b/Scripts/evaluate_neural_network.py index dd4a09f..97d343e 100644 --- a/Scripts/evaluate_neural_network.py +++ b/Scripts/evaluate_neural_network.py @@ -44,6 +44,7 @@ def generate_and_save_comparison(abs_path): evaluation_file_path = os.path.join(abs_path, 'neural_network_evaluation.csv') df = pd.read_csv(evaluation_file_path, sep=',', header=None, names=['build_number', 'Accuracy', 'Loss', 'Precision', 'Recall', 'F1']) + df = df[df.build_number != 0] fig = plt.figure(figsize=(16 * .6, 9 * .6)) ax = fig.add_subplot(111) ax.set_title('Evaluation')