s464980/IUM_s464980
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix metric to rmse bc continuous values xd
All checks were successful
s464980-evaluation/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Sheaza 2024-05-14 22:23:41 +02:00
parent df124bcad7
commit 7cee641b3b
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
predict.py
View File

@ -1,7 +1,7 @@
import pandas as pd import pandas as pd
from tensorflow import keras from tensorflow import keras
import numpy as np import numpy as np
from sklearn.metrics import accuracy_score from sklearn.metrics import root_mean_squared_error
np.set_printoptions(threshold=np.inf) np.set_printoptions(threshold=np.inf)
data = pd.read_csv("df_test.csv") data = pd.read_csv("df_test.csv")
@ -15,6 +15,6 @@ predictions = model.predict(X_test)
with open("predictions.txt", "w") as f: with open("predictions.txt", "w") as f:
f.write(str(predictions)) 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: with open("accuracy.txt", 'a') as file:
file.write(str(accuracy)) file.write(str(accuracy))