From 41f8394f8fdb34755b22b4bead7641eff99b892a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebastian=20Wa=C5=82=C4=99sa?= Date: Sun, 15 May 2022 22:08:21 +0200 Subject: [PATCH] Zaktualizuj 'ml_pytorch_mlflow.py' --- ml_pytorch_mlflow.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/ml_pytorch_mlflow.py b/ml_pytorch_mlflow.py index 37979a2..ed3796f 100644 --- a/ml_pytorch_mlflow.py +++ b/ml_pytorch_mlflow.py @@ -130,10 +130,12 @@ def my_main(epochs): expected = [] predicted = [] + inputs = [] for i in range(0, len(val_ds), 1): input_, target = val_ds[i] expected.append(float(target)) predicted.append(float(prediction(input_, model))) + inputs.append(input_) MSE = mean_squared_error(expected, predicted) MAE = mean_absolute_error(expected, predicted) @@ -147,7 +149,7 @@ def my_main(epochs): input_, target = val_ds[i] file.write(str(predict_single(input_, target, model))) - print(val_ds[0]) + print(inputs) # input_example = val_ds[0] # # input_example = input_example.unsqueeze(0)