diff --git a/scripts/predict_s444356.py b/scripts/predict_s444356.py index dd5044d..7676683 100644 --- a/scripts/predict_s444356.py +++ b/scripts/predict_s444356.py @@ -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))}") diff --git a/scripts/predict_s444356_registry.py b/scripts/predict_s444356_registry.py index b261a10..3691916 100644 --- a/scripts/predict_s444356_registry.py +++ b/scripts/predict_s444356_registry.py @@ -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))}")