From bf1f1129fa97f8b4694df667895ffbfad9104fb4 Mon Sep 17 00:00:00 2001 From: Dawid Date: Sun, 23 May 2021 17:52:53 +0200 Subject: [PATCH] change model name --- mlflow_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlflow_model.py b/mlflow_model.py index 7d974ba..32c165f 100644 --- a/mlflow_model.py +++ b/mlflow_model.py @@ -57,7 +57,7 @@ if __name__ == "__main__": mlf.set_experiment("s434804") tracking_url_type_store = urlparse(mlf.get_tracking_uri()).scheme if tracking_url_type_store != "file": - mlf.keras.log_model(model, "country_vaccinations", registered_model_name="s434804", signature=signature, + mlf.keras.log_model(model, "country_vaccinations2", registered_model_name="s434804", signature=signature, input_example=input_example) else: mlf.keras.log_model(model, "vaccines_model", signature=signature, input_example=input_example)