s444517
/
ium_444517
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

mlflow attempt task 2

This commit is contained in:
Kamila 2022-05-15 12:04:04 +02:00
parent bfc88a043a
commit 7ab7456488
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nn_train_mlflow.py
View File

@ -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)
# mlflow.keras.save_model(model, "my_model_mlflow", signature=signature, input_example=input_example)