added labels to evaluations
All checks were successful
s444354-training/pipeline/head This commit looks good
s444354-evaluation/pipeline/head This commit looks good

This commit is contained in:
Adrian Charkiewicz 2022-05-07 05:25:40 +02:00
parent 192e07685b
commit d8abc6f7c2

View File

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