s434788/ium_434788
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

gwegwg

Browse Source
This commit is contained in:
Dominik Strzako 2021-05-27 12:48:20 +02:00
parent c0328b8925
commit 898551d710
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
Zadanie_09_MLflow_Predict_Registry.py
View File

@ -5,11 +5,15 @@ import pandas as pd
mlflow.set_tracking_uri("http://172.17.0.1:5000") mlflow.set_tracking_uri("http://172.17.0.1:5000")
client = MlflowClient() client = MlflowClient()
version = 3 version = 3
model_name = "s434788" model_name = "s434695"
#for mv in client.search_model_versions(f"name='{model_name}'"): #for mv in client.search_model_versions(f"name='{model_name}'"):
# if int(mv.version) > version: # if int(mv.version) > version:
# version = int(mv.version) # version = int(mv.version)
print(f"selected version {version}")
print(f"I will search for URI: models:/{model_name}/{version}")
model = mlflow.keras.load_model( model = mlflow.keras.load_model(
model_uri=f"models:/{model_name}/{version}" model_uri=f"models:/{model_name}/{version}"
) )