mlflow attempt task 2 prediction

2022-05-15 12:59:43 +02:00 · 2022-05-15 12:59:43 +02:00 · 4380b0c24b
commit 4380b0c24b
parent 484aa77a51
1 changed files with 1 additions and 1 deletions
--- 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))}")
+    print(f"Prediction: {model.predict(np.array([float(json_data['inputs'])]))}")