added labels to evaluations
This commit is contained in:
parent
192e07685b
commit
d8abc6f7c2
@ -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')
|
Loading…
Reference in New Issue
Block a user