s434784/ium_434784
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

sacred metrics

Browse Source
This commit is contained in:
Maciej Sobkowiak 2021-05-17 00:46:04 +02:00
parent 670b605789
commit 15ac30cf27
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
sacred_training.py
View File

@ -58,6 +58,7 @@ def prepare(epochs, batch_size, _run):
predictions = model.predict(X_test) predictions = model.predict(X_test)
error = mean_squared_error(y_test, predictions) error = mean_squared_error(y_test, predictions)
_run.info["mean_squared_error"] = str(error) _run.info["mean_squared_error"] = str(error)
_run.log_scalar("mean_squared_error", int(error))
return error return error