fix metric to rmse bc continuous values xd
All checks were successful
s464980-evaluation/pipeline/head This commit looks good
All checks were successful
s464980-evaluation/pipeline/head This commit looks good
This commit is contained in:
parent
df124bcad7
commit
7cee641b3b
@ -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))
|
||||
|
Loading…
Reference in New Issue
Block a user