13 lines
320 B
Python
13 lines
320 B
Python
import numpy as np
|
|
import json
|
|
import mlflow
|
|
import sys
|
|
|
|
|
|
model = 'mlruns/1/70439eb482b54d56b54b0ecc6f1ca96f/artifacts/s444409'
|
|
model = mlflow.pyfunc.load_model(model)
|
|
|
|
example = sys.argv[1]
|
|
data_p = np.array([example['inputs'][0]], dtype=np.float32)
|
|
print(10*'=' + 'PREDICTIONS' + 10*'=')
|
|
print({model.predict(data_p)}) |