From 9b2764a4c0cb7d508db45054cb0a78be5c033fb9 Mon Sep 17 00:00:00 2001 From: patrycjalazna Date: Sat, 15 May 2021 13:18:37 +0200 Subject: [PATCH] added rmse to sacred --- sacred-fileobserver.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/sacred-fileobserver.py b/sacred-fileobserver.py index e9900b8..48628d3 100644 --- a/sacred-fileobserver.py +++ b/sacred-fileobserver.py @@ -23,7 +23,7 @@ def my_config(): batch_size = 16 @ex.capture -def prepare_model(epochs, batch_size): +def prepare_model(epochs, batch_size, _run): # odczytanie danych z plików avocado_train = pd.read_csv('avocado_train.csv') avocado_test = pd.read_csv('avocado_test.csv') @@ -55,6 +55,7 @@ def prepare_model(epochs, batch_size): # ewaluacja rmse = mean_squared_error(y_test, prediction) + _run.log_scalar("rmse", rmse) # zapisanie modelu model.save('avocado_model.h5')