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

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

Browse Source
This commit is contained in:
Kacper Dudzic 2022-05-14 13:22:33 +02:00
parent 0559fd6038
commit 1d74959d77
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

@ -89,7 +89,7 @@ def train():
tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
if tracking_url_type_store != 'file': if tracking_url_type_store != 'file':
mlflow.keras.log_model(model, 'lego-model', registered_model_name='TFLegoModel', mlflow.keras.log_model(model, 'lego-model', registered_model_name='TFLegoModel',
signature=signature) signature=signature, input_example=np.array(500))
else: else:
mlflow.keras.log_model(model, 'model', signature=signature, input_example=np.array(500)) mlflow.keras.log_model(model, 'model', signature=signature, input_example=np.array(500))