From 15ac30cf2762a12c7a1c1f4716bcbaa92101de19 Mon Sep 17 00:00:00 2001
From: Maciej Sobkowiak <masobkowiak@gmail.com>
Date: Mon, 17 May 2021 00:46:04 +0200
Subject: [PATCH] sacred metrics

---
 sacred_training.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/sacred_training.py b/sacred_training.py
index 1eb1db1..252bfd9 100644
--- 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