From bcaba678d379b857b46edcb5742787b8ffdbb344 Mon Sep 17 00:00:00 2001 From: s444417 Date: Sun, 15 May 2022 10:57:42 +0200 Subject: [PATCH] reshape --- lab8/predictMlflow.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lab8/predictMlflow.py b/lab8/predictMlflow.py index 6ab7187..ed6b8e6 100644 --- a/lab8/predictMlflow.py +++ b/lab8/predictMlflow.py @@ -24,7 +24,7 @@ model = mlflow.pyfunc.load_model(registry_path) with open(f'{registry_path}/input_example.json') as f: input_example_data = json.load(f) -input_example = np.array(input_example_data['inputs']) +input_example = np.array(input_example_data['inputs']).reshape(-1, 1) print(f'Input: {input_example}') print(f'Prediction: {model.predict(input_example)}') \ No newline at end of file