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

mlflow attempt task 2 prediction

Browse Source
This commit is contained in:
Kamila 2022-05-15 12:58:16 +02:00
parent e9ebc9a95b
commit 484aa77a51
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
predict.py
View File

@ -14,5 +14,5 @@ model =mlflow.pyfunc.load_model(f"{PATH}")
with open(f'{PATH}{input}', 'r') as file:
json_data = json.load(file)
print(f"Input: {json_data['inputs'][0]}")
print(f"Prediction: {model.predict(np.array([json_data['inputs'][0]], dtype=np.float32))}")
print(f"Input: {json_data['inputs']}")
print(f"Prediction: {model.predict(np.array([json_data['inputs']], dtype=np.float32))}")