s430705/ium_430705
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Usuń 'lab09_02_predict.py'

Browse Source
This commit is contained in:
Michał Zaręba 2021-06-12 16:53:08 +02:00
parent aeb428fbbd
commit cd2500c9b6
1 changed files with 0 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
lab09_02_predict.py
View File

@ -1,21 +0,0 @@
import json
import mlflow
import pandas as pd
import mlflow.pyfunc
model_name = "country_vaccination"
stage = '1'
model = mlflow.pyfunc.load_model(
model_uri=f"models:/{model_name}/{stage}"
)
with open('country_vaccination/input_example.json', 'r') as datafile:
data = json.load(datafile)
example_input = data["inputs"]
input_dictionary = {i : x for i, x in enumerate(example_input) }
input_ex = pd.DataFrame(input_dictionary, index=[0])
print(model.predict(input_ex))