added legend
This commit is contained in:
parent
d8abc6f7c2
commit
272cd623ff
@ -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')
|
Loading…
Reference in New Issue
Block a user