From 7ab7456488aff61fafef430a7cb54c3f1177eb42 Mon Sep 17 00:00:00 2001 From: Kamila Date: Sun, 15 May 2022 12:04:04 +0200 Subject: [PATCH] mlflow attempt task 2 --- nn_train_mlflow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nn_train_mlflow.py b/nn_train_mlflow.py index 803c35c..f76ab11 100644 --- a/nn_train_mlflow.py +++ b/nn_train_mlflow.py @@ -14,7 +14,7 @@ from urllib.parse import urlparse mlflow.set_tracking_uri("http://172.17.0.1:5000") mlflow.set_experiment("s_444517") - +# w nazwie dodalam podkreslnik poniewaz dostawalam bardzo dziwny blad o istniejacym eksperymencie ze statusem "deleted" i nie udalo mi sie przywrocic oryginalnego eksperymentu # reading data def read_data(): @@ -106,4 +106,4 @@ with mlflow.start_run(): mlflow.sklearn.log_model(model, "my_model_mlflow", registered_model_name="s_444517", signature=signature, input_example=input_example) else: mlflow.sklearn.log_model(model, "my_model_mlflow", registered_model_name="s_444517", signature=signature, input_example=input_example) - mlflow.keras.save_model(model, "my_model_mlflow", signature=signature, input_example=input_example) \ No newline at end of file + # mlflow.keras.save_model(model, "my_model_mlflow", signature=signature, input_example=input_example) \ No newline at end of file