From de2b7054e98a14c4b042ab21fa652cf0e71412f5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebastian=20Wa=C5=82=C4=99sa?= Date: Sun, 15 May 2022 22:33:09 +0200 Subject: [PATCH] Zaktualizuj 'ml_pytorch_mlflow.py' --- ml_pytorch_mlflow.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ml_pytorch_mlflow.py b/ml_pytorch_mlflow.py index 62e684a..f60476a 100644 --- a/ml_pytorch_mlflow.py +++ b/ml_pytorch_mlflow.py @@ -135,10 +135,10 @@ def my_main(epochs): input_, target = val_ds[i] expected.append(float(target)) predicted.append(float(prediction(input_, model))) - input_ = pd.array(float(input_)) + input_ = pd.array(input_) inputs.append(input_.numpy()) - inputs = pd.DataFrame(inputs) + # inputs = pd.DataFrame(inputs) MSE = mean_squared_error(expected, predicted) MAE = mean_absolute_error(expected, predicted)