diff --git a/predict.py b/predict.py index c25933b..4ca07a0 100644 --- a/predict.py +++ b/predict.py @@ -14,8 +14,8 @@ y_to_test = pandas.read_csv('./Y_test.csv') accu = reloaded.evaluate(x_to_test, y_to_test) -with open('metrics.txt', 'a') as file: - file.write(f'Accuracy: {accu}') +with open('metrics.csv', 'a') as file: + file.write(f'{accu},') pre = reloaded.predict(x_to_test)