kdudzic/ium_s449288
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update 'simple_regression_lab8.py'
Some checks failed
s449288-evaluation/pipeline/head There was a failure building this commit
Details
s449288-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Kacper Dudzic 2022-05-13 10:40:03 +02:00
parent 14c5aacb08
commit 59b8d51165
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
simple_regression_lab8.py
View File

@ -112,7 +112,7 @@ def train(epochs, units, learning_rate, _run):
mlflow.keras.log_model(model, 'lego-model', registered_model_name='TFLegoModel',
signature=signature)
else:
mlflow.keras.log_model(model, 'model', signature=signature, input_example=500)
mlflow.keras.log_model(model, 'model', signature=signature, input_example=np.array(500))
@ex.automain