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')