diff --git a/predict.py b/predict.py index 65e559e..3ac3c8c 100644 --- a/predict.py +++ b/predict.py @@ -16,5 +16,5 @@ with open("predictions.txt", "w") as f: f.write(str(predictions)) accuracy = root_mean_squared_error(y_test, predictions) -with open("accuracy.txt", 'a') as file: +with open("rmse.txt", 'a') as file: file.write(str(accuracy))