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

mlflow attempt task 2 prediction registry

Browse Source
This commit is contained in:
Kamila 2022-05-15 13:30:37 +02:00
parent 7baecd43ef
commit ce8188b06b
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
predict_registry.py
View File

@ -4,12 +4,16 @@ import numpy as np
import sys
PATH = "mlruns/14/80fe21a0804844088147d15a3cebb3e5/artifacts/lego-model"
PATH = "/mlruns/14/80fe21a0804844088147d15a3cebb3e5/artifacts/lego-model"
model_version = 14
model_name = "s449288"
model = mlflow.pyfunc.load_model(PATH)
model = mlflow.pyfunc.load_model(
model_uri=f"models:/{model_name}/{model_version}"
)
with open(f'{PATH}/{(sys.argv[1:])[0]}', 'r') as file:
json_data = json.load(file)
print(f"Prediction: {model.predict(np.array([json_data['inputs']]))}")
print(f"Prediction: {model.predict(np.array([500]))}")