From 47c8e5fd21c4e9c0a8a8ef2f49b25303f963703a Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Fri, 5 Feb 2021 17:31:09 +0100
Subject: [PATCH] Predictions are now correctly aggregated to numpy array.
---
pysegcnn/core/trainer.py | 6 ++++--
1 file changed, 4 insertions(+), 2 deletions(-)
diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 50b91d6..4d41611 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2706,8 +2706,10 @@ class NetworkInference(BaseConfig):
str(base_name.name).replace(fold_number, 'kfold'))
# predictions of the different models
- y_true = [output['y_true'] for output in inference.values()]
- y_pred = [output['y_pred'] for output in inference.values()]
+ y_true = np.asarray(
+ [output['y_true'] for output in inference.values()]).flatten()
+ y_pred = np.asarray(
+ [output['y_pred'] for output in inference.values()]).flatten()
# calculate classification over all different models
LOGGER.info('Aggregating statistics of models:')
--
GitLab