From 4380b0c24b8ded35704519c0ee0bc12fa4a59f04 Mon Sep 17 00:00:00 2001 From: Kamila Date: Sun, 15 May 2022 12:59:43 +0200 Subject: [PATCH] mlflow attempt task 2 prediction --- predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/predict.py b/predict.py index 6929115..bd9e818 100644 --- a/predict.py +++ b/predict.py @@ -15,4 +15,4 @@ with open(f'{PATH}{input}', 'r') as file: json_data = json.load(file) print(f"Input: {json_data['inputs']}") - print(f"Prediction: {model.predict(np.array([json_data['inputs']], dtype=np.float32))}") \ No newline at end of file + print(f"Prediction: {model.predict(np.array([float(json_data['inputs'])]))}") \ No newline at end of file