added rmse to sacred

2021-05-15 13:18:37 +02:00 · 2021-05-15 13:18:37 +02:00 · 9b2764a4c0
commit 9b2764a4c0
parent 6596e3373e
1 changed files with 2 additions and 1 deletions
--- 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')