s444409/ium_444409
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add scripts to use someones model

Browse Source
This commit is contained in:
Marcin Kostrzewski 2022-05-11 18:34:08 +02:00
parent 39acc4bd27
commit 5db367b894
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
predict_s444356.py
View File

@ -1,8 +1,6 @@
import json
import mlflow
import numpy as np
import torch
from torch.autograd import Variable
logged_model = 'mlruns/1/d5b6f9c1784a4d2dbb8592cd4ad364d7/artifacts/model'
loaded_model = mlflow.pyfunc.load_model(logged_model)
@ -10,7 +8,7 @@ loaded_model = mlflow.pyfunc.load_model(logged_model)
with open(f'{logged_model}/input_example.json') as f:
data = json.load(f)
input_example = np.array(data['inputs'][0])
input_example = Variable(torch.from_numpy(input_example)).float()
input_example = np.array([data['inputs'][0]], dtype=np.float64).reshape(-1, 2)
loaded_model.predict(input_example)
print(f'Prediction: {loaded_model.predict(input_example)}')