From af9ed1dc49138eb5cb2507b754280975d2f321d9 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Fri, 5 Feb 2021 17:13:53 +0100
Subject: [PATCH] Debugged classification report logging.
---
pysegcnn/core/trainer.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 7df3bca..9501715 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2618,9 +2618,6 @@ class NetworkInference(BaseConfig):
dictConfig(log_conf(log.log_file))
log.init_log('Evaluating model: {}.'.format(state))
- # classification report labels
- cr_labels = [v['label'] for _, v in self.source_labels.items()]
-
# check whether model was already evaluated
if self.eval_file(state).exists():
LOGGER.info('Found existing model evaluation: {}.'
@@ -2663,6 +2660,9 @@ class NetworkInference(BaseConfig):
output['y_true'] = y_true
output['y_pred'] = y_pred
+ # classification report labels
+ cr_labels = [v['label'] for _, v in self.source_labels.items()]
+
# calculate classification report from sklearn
report_name = self.report_path.joinpath(self.report_name(state))
LOGGER.info('Calculating classification report: {}'
--
GitLab