diff --git a/predict.py b/predict.py index 1f4efd1..8323c74 100644 --- a/predict.py +++ b/predict.py @@ -14,7 +14,7 @@ y_to_test = pandas.read_csv('./Y_test.csv') accu = reloaded.evaluate(x_to_test, y_to_test) -with open('metrics.py', 'w') as file: +with open('metrics.txt', 'w') as file: file.write(f'accurency: {accu}') pre = reloaded.predict(x_to_test)