From d5d573f1988b4c7dfa7f34e621a2ce32d510d553 Mon Sep 17 00:00:00 2001 From: michalzareba Date: Sun, 23 May 2021 17:14:19 +0200 Subject: [PATCH] Add model registry --- lab08_mfl.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lab08_mfl.py b/lab08_mfl.py index 01d7b88..b438d49 100644 --- a/lab08_mfl.py +++ b/lab08_mfl.py @@ -86,7 +86,7 @@ with mlflow.start_run(): tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme 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) else: mlflow.keras.log_model(model, "model_movies", signature=signature, input_example=input_example)