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

one more time

Browse Source
This commit is contained in:
Dominik Strzako 2021-05-24 01:01:49 +02:00
parent 315a3d3bda
commit aa78c08876
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Zadanie_09_MLflow_Predict_Registry.py
View File

@ -10,7 +10,7 @@ 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)
model = mlflow.pytorch.load_model( model = mlflow.keras.load_model(
model_uri=f"models:/{model_name}/{version}" model_uri=f"models:/{model_name}/{version}"
) )