sacred metrics

2021-05-17 00:46:04 +02:00 · 2021-05-17 00:46:04 +02:00 · 15ac30cf27
commit 15ac30cf27
parent 670b605789
1 changed files with 1 additions and 0 deletions
--- a/sacred_training.py
+++ b/sacred_training.py
@ -58,6 +58,7 @@ def prepare(epochs, batch_size, _run):
    predictions = model.predict(X_test)
    error = mean_squared_error(y_test, predictions)
    _run.info["mean_squared_error"] = str(error)
+    _run.log_scalar("mean_squared_error", int(error))
    return error