s434700/ium_s434700
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Testing MlflowClient

Browse Source
This commit is contained in:
Filip Izydorczyk 2021-06-10 14:45:52 +02:00
parent 805da075ae
commit bda0aec524
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
predict-registry/predict.py
View File

@ -12,3 +12,6 @@ mlflow.set_tracking_uri("http://172.17.0.1:5000")
model = mlflow.keras.pyfunc.load_model( model = mlflow.keras.pyfunc.load_model(
model_uri=f"models:/{model_name}/{model_version}", model_uri=f"models:/{model_name}/{model_version}",
) )
client = MlflowClient()
print(client.get_model_version(model_name, model_version).source)