kdudzic/ium_s449288
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update 'predict_registry.py'

Browse Source
This commit is contained in:
Kacper Dudzic 2022-05-14 14:56:26 +02:00
parent eddfb61cfc
commit c235cf56ff
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
predict_registry.py
View File

@ -2,13 +2,13 @@ import mlflow
import numpy as np
import json
registry_path = '/mlruns/17/4a8894c60fe34adcb79108c66d7330fc/artifacts/linear-model' #
model = mlflow.pyfunc.load_model(registry_path) #
registry_path = '/mlruns/17/4a8894c60fe34adcb79108c66d7330fc/artifacts/linear-model'
model = mlflow.pyfunc.load_model(registry_path)
with open(f'{registry_path}/input_example.json') as f:
input_example_data = json.load(f)
input_example = np.array(input_example_data['inputs']).reshape(-1, 8) #
input_example = np.array(input_example_data['inputs']).reshape(-1, 8)
print(f'Input example: {input_example}')
print(f'Model prediction: {model.predict(input_example)}')