Update 'simple_regression_lab7.py'

2022-05-05 23:13:22 +02:00 · 2022-05-05 23:13:22 +02:00 · 7665cbeea5
commit 7665cbeea5
parent c48ab0ae5c
1 changed files with 1 additions and 1 deletions
--- a/simple_regression_lab7.py
+++ b/simple_regression_lab7.py
@ -74,7 +74,7 @@ def train(epochs, units, learning_rate, _run):

    # Zapis modelu do pliku standardowo poprzez metodę kerasa i poprzez metodę obiektu Experiment z Sacred
    model.save('lego_reg_model')
-    ex.add_artifact('lego_reg_model_art')
+    ex.add_artifact('lego_reg_model')

    # Przykładowo zwracamy coś w charakterze wyników, żeby było widoczne w plikach zapisanych przez obserwator
    hist = pd.DataFrame(history.history)