From 272cd623ff84bc65efc617dad76bbea40efe4358 Mon Sep 17 00:00:00 2001
From: Adrian Charkiewicz <riraasaa@gmail.com>
Date: Sat, 7 May 2022 05:30:34 +0200
Subject: [PATCH] added legend

---
 evaluation.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/evaluation.py b/evaluation.py
index aea3416..f8f7b0e 100644
--- a/evaluation.py
+++ b/evaluation.py
@@ -38,10 +38,9 @@ with open('eval.csv', 'r') as r:
         MSE.append(float(row[1]))        
         RMSE.append(float(row[2]))    
 
-
-plt.ylabel('F score')
+plt.xlabel('build')
 plt.plot(np.arange(0, len(MAE)), MAE, label="MAE")
 plt.plot(np.arange(0, len(MSE)), MSE, label="MSE") 
 plt.plot(np.arange(0, len(RMSE)), RMSE, label="RMSE") 
-
+plt.legend()
 plt.savefig('metrics.png')
\ No newline at end of file