s430705/ium_430705
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add model registry
Some checks failed
s430705-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
michalzareba 2021-05-23 17:14:19 +02:00
parent 6549b4dc98
commit d5d573f198
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lab08_mfl.py
View File

@ -86,7 +86,7 @@ with mlflow.start_run():
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, "movies_imdb", registered_model_name="s430705", signature=signature, mlflow.keras.log_model(model, "movies_imdb2", registered_model_name="s430705", signature=signature,
input_example=input_example) input_example=input_example)
else: else:
mlflow.keras.log_model(model, "model_movies", signature=signature, input_example=input_example) mlflow.keras.log_model(model, "model_movies", signature=signature, input_example=input_example)