diff --git a/predict.py b/predict.py index 990305e..65e559e 100644 --- a/predict.py +++ b/predict.py @@ -1,7 +1,7 @@ import pandas as pd from tensorflow import keras import numpy as np -from sklearn.metrics import accuracy_score +from sklearn.metrics import root_mean_squared_error np.set_printoptions(threshold=np.inf) data = pd.read_csv("df_test.csv") @@ -15,6 +15,6 @@ predictions = model.predict(X_test) with open("predictions.txt", "w") as f: f.write(str(predictions)) -accuracy = accuracy_score(y_test, predictions) +accuracy = root_mean_squared_error(y_test, predictions) with open("accuracy.txt", 'a') as file: file.write(str(accuracy))