ium_478841/scripts/predict_s444356.py
2022-05-15 12:44:15 +02:00

11 lines
318 B
Python

import mlflow
import numpy as np
import json
model = mlflow.pyfunc.load_model(
'/mlruns/13/da5c6167bb45403fa35569849a1fbc13/artifacts/model')
with open('input_example.json') as f:
data = np.array([json.load(f)['inputs'][0]], dtype=np.float32)
print(f"Predicted values: {model.predict(data.reshape(-1, 2))}")