From d8abc6f7c2e97261b2809e920bfcb1799cb7f2b9 Mon Sep 17 00:00:00 2001 From: Adrian Charkiewicz Date: Sat, 7 May 2022 05:25:40 +0200 Subject: [PATCH] added labels to evaluations --- evaluation.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/evaluation.py b/evaluation.py index c4ea784..aea3416 100644 --- a/evaluation.py +++ b/evaluation.py @@ -40,8 +40,8 @@ with open('eval.csv', 'r') as r: plt.ylabel('F score') -plt.plot(np.arange(0, len(MAE)), MAE) -plt.plot(np.arange(0, len(MSE)), MSE) -plt.plot(np.arange(0, len(RMSE)), RMSE) +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.savefig('metrics.png') \ No newline at end of file