s495719/ium_495719
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix sacred

Browse Source
This commit is contained in:
PawelDopierala 2024-06-12 14:58:14 +02:00
parent a305b06012
commit 7a813295f2
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
sacred/create_model.py
View File

@ -53,9 +53,6 @@ def main(epochs, learning_rate, batch_size, _run):
test_predictions = model.predict(X_test)
predictions_df = pd.DataFrame(test_predictions, columns=["Predicted_Price"])
predictions_df.to_csv('hp_test_predictions.csv', index=False)
rmse = np.sqrt(mean_squared_error(Y_test, test_predictions))
mae = mean_absolute_error(Y_test, test_predictions)
r2 = r2_score(Y_test, test_predictions)