added legend
s444354-training/pipeline/head This commit looks good Details
s444354-evaluation/pipeline/head This commit looks good Details

This commit is contained in:
Adrian Charkiewicz 2022-05-07 05:30:34 +02:00
parent d8abc6f7c2
commit 272cd623ff
1 changed files with 2 additions and 3 deletions

View File

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