From 2c3dc7cc0b26883461d51d1bb6156d3db5979439 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebastian=20Wa=C5=82=C4=99sa?= Date: Sun, 15 May 2022 22:42:43 +0200 Subject: [PATCH] Zaktualizuj 'ml_pytorch_mlflow.py' --- ml_pytorch_mlflow.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ml_pytorch_mlflow.py b/ml_pytorch_mlflow.py index 63aa352..474c72e 100644 --- a/ml_pytorch_mlflow.py +++ b/ml_pytorch_mlflow.py @@ -137,7 +137,8 @@ def my_main(epochs): predicted.append(float(prediction(input_, model))) inputs.append(input_.numpy()) - inputs = inputs.to_array() + inputs = pd.DataFrame(inputs, dtype=np.float64) + inputs = inputs.to_numpy() MSE = mean_squared_error(expected, predicted) MAE = mean_absolute_error(expected, predicted)