diff --git a/scripts/evaluate.py b/scripts/evaluate.py index c72cbef..29928ac 100644 --- a/scripts/evaluate.py +++ b/scripts/evaluate.py @@ -20,7 +20,8 @@ with open('./data/evaluation_results.csv', 'a+') as f: f.write("{0},{1},{2}\n".format(*evaluate_model(test_data, model))) # * Load all inference data gathered (till the current one) -results = pd.read_csv('./data/evaluation_results.csv', header=0) +results = pd.read_csv('./data/evaluation_results.csv', + names=['MSE', 'RMSE', 'MAE']) # * Plot the results plt.plot(range(1, len(results)+1), results['MSE'], color='green')