From f5175a9517ab8bdea06c75b55de264ed2dc93061 Mon Sep 17 00:00:00 2001 From: s444417 Date: Sun, 15 May 2022 11:05:05 +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 ed6b8e6..cadfe57 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']).reshape(-1, 1) +input_example = np.array(input_example_data['inputs']).reshape(-1,) print(f'Input: {input_example}') print(f'Prediction: {model.predict(input_example)}') \ No newline at end of file