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

Print experiment info
Some checks failed
s434704-predict-s426206-from-registry/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Wojciech Jarmosz 2021-05-23 23:28:29 +02:00
parent 797e893eda
commit d2014bef6a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mlflow_prediction_registry.py
View File

@ -22,7 +22,7 @@ experiment = client.get_latest_versions(model_name, stages=None)
print(experiment) print(experiment)
with open(f'{model_name}/{input}', 'r') as file: with open(f'{experiment.source}/{input}', 'r') as file:
json_data = json.load(file) json_data = json.load(file)
print(model(torch.tensor(np.array(json_data['inputs'])).float())) print(model(torch.tensor(np.array(json_data['inputs'])).float()))