From 22b7a65a1738b82796cb2e87864b0da648a4ee9c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebastian=20Wa=C5=82=C4=99sa?= Date: Sun, 15 May 2022 23:02:40 +0200 Subject: [PATCH] Zaktualizuj 'ml_pytorch_mlflow.py' --- ml_pytorch_mlflow.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/ml_pytorch_mlflow.py b/ml_pytorch_mlflow.py index 69c8100..d979019 100644 --- a/ml_pytorch_mlflow.py +++ b/ml_pytorch_mlflow.py @@ -132,14 +132,17 @@ def my_main(epochs): expected = [] predicted = [] inputss = [] + targetss = [] for i in range(0, len(val_ds), 1): input_, target = val_ds[i] expected.append(float(target)) predicted.append(float(prediction(input_, model))) inputss.append(input_) - inputss = pd.DataFrame(inputss, dtype=np.float64) - inputss = inputss.to_numpy() + + inputss, targetss = val_ds + # inputss = pd.DataFrame(inputss, dtype=np.float64) + # inputss = inputss.to_numpy() MSE = mean_squared_error(expected, predicted) MAE = mean_absolute_error(expected, predicted)