Dtype fix

This commit is contained in:
MatOgr 2022-05-15 12:59:27 +02:00
parent 6515322f1c
commit 2d12bf9ced
2 changed files with 2 additions and 2 deletions

View File

@ -6,5 +6,5 @@ import json
model = mlflow.pyfunc.load_model(
'mlruns/1/4b83e774512444188fb587288818c298/artifacts/model')
with open('scripts/input_example.json') as f:
data = np.array([json.load(f)['inputs'][0]], dtype=np.float32)
data = np.array([json.load(f)['inputs'][0]], dtype=np.float64)
print(f"Predicted values: {model.predict(data.reshape(-1, 2))}")

View File

@ -8,5 +8,5 @@ print('################\n\n', os.listdir(), "\n\n###############")
model = mlflow.pyfunc.load_model(
'mlruns/13/da5c6167bb45403fa35569849a1fbc13/artifacts/model')
with open('mlruns/13/da5c6167bb45403fa35569849a1fbc13/artifacts/model/input_example.json') as f:
data = np.array([json.load(f)['inputs'][0]], dtype=np.float32)
data = np.array([json.load(f)['inputs'][0]], dtype=np.float64)
print(f"Predicted values: {model.predict(data.reshape(-1, 2))}")