s434704/ium_434704
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add logs
All checks were successful
s434704-predict-s426206/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Wojciech Jarmosz 2021-05-23 18:57:18 +02:00
parent 99aa9a85a0
commit c221c48a4a
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mlflow_prediction.py
View File

@ -14,6 +14,9 @@ input = str(arguments[0])
model = mlflow.pytorch.load_model("my_model") model = mlflow.pytorch.load_model("my_model")
with open(f'my_model/{input}', 'r') as file: with open(f'my_model/{input}', 'r') as file:
json_data = json.load(file)['inputs'] json_data = json.load(file)
print(json_data)
print(json_data['inputs'])
print(model(torch.tensor(np.array(json_data['inputs'])).float())) print(model(torch.tensor(np.array(json_data['inputs'])).float()))