From 7cee641b3b35896688926339751ae01d193bd1a7 Mon Sep 17 00:00:00 2001 From: Sheaza Date: Tue, 14 May 2024 22:23:41 +0200 Subject: [PATCH] fix metric to rmse bc continuous values xd --- predict.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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))