Update 'simple_regression_lab7.py'
All checks were successful
s449288-evaluation/pipeline/head This commit looks good
s449288-training/pipeline/head This commit looks good

This commit is contained in:
Kacper Dudzic 2022-05-05 23:13:22 +02:00
parent c48ab0ae5c
commit 7665cbeea5

View File

@ -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)