2022-05-15 12:33:21 +02:00
|
|
|
import mlflow
|
|
|
|
|
import numpy as np
|
|
|
|
|
import json
|
2022-05-15 12:50:04 +02:00
|
|
|
import os
|
2022-05-15 12:33:21 +02:00
|
|
|
|
|
|
|
|
|
2022-05-15 12:50:04 +02:00
|
|
|
print('################\n\n', os.listdir(), "\n\n###############")
|
2022-05-15 12:33:21 +02:00
|
|
|
model = mlflow.pyfunc.load_model(
|
2022-05-15 12:54:51 +02:00
|
|
|
'mlruns/1/4b83e774512444188fb587288818c298/artifacts/model')
|
|
|
|
|
with open('mlruns/1/4b83e774512444188fb587288818c298/artifacts/model/input_example.json') as f:
|
2022-05-15 12:33:21 +02:00
|
|
|
data = np.array([json.load(f)['inputs'][0]], dtype=np.float32)
|
|
|
|
|
print(f"Predicted values: {model.predict(data.reshape(-1, 2))}")
|
