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 15:08:00 +02:00
parent 7a813295f2
commit 408bebce7a
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
sacred/create_model.py
View File

@ -55,11 +55,9 @@ def main(epochs, learning_rate, batch_size, _run):
rmse = np.sqrt(mean_squared_error(Y_test, test_predictions)) rmse = np.sqrt(mean_squared_error(Y_test, test_predictions))
mae = mean_absolute_error(Y_test, test_predictions) mae = mean_absolute_error(Y_test, test_predictions)
r2 = r2_score(Y_test, test_predictions)
_run.log_scalar("rmse", rmse) _run.log_scalar("rmse", rmse)
_run.log_scalar("mae", mae) _run.log_scalar("mae", mae)
_run.log_scalar("r2", r2)
if __name__ == '__main__': if __name__ == '__main__':