fix
Some checks failed
s444452-training/pipeline/head This commit looks good
s444452-evaluation/pipeline/head There was a failure building this commit

This commit is contained in:
AdamOsiowy123 2022-05-04 20:41:25 +02:00
parent eab44fadf4
commit c5565d64fd

View File

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