update predict.py
All checks were successful
s444356-evaluation/pipeline/head This commit looks good
s444356-training/pipeline/head This commit looks good

This commit is contained in:
Maciej Czajka 2022-05-11 20:13:00 +02:00
parent eda722bcfd
commit 6dbfc61a3c

View File

@ -13,4 +13,4 @@ with open(f'{logged_model}/'+input) as f:
data = json.load(f) data = json.load(f)
input_example = np.array([data['inputs']], dtype=np.float64) input_example = np.array([data['inputs']], dtype=np.float64)
loaded_model.predict(data['inputs']) loaded_model.predict(input_example)