From d1875895582ebcc466cc6a271445801b4a0bac85 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Fri, 5 Feb 2021 17:16:49 +0100
Subject: [PATCH] Debugged classification report logging.
---
pysegcnn/core/trainer.py | 6 +++---
pysegcnn/core/utils.py | 2 +-
2 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 9501715..f23a72a 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2583,8 +2583,7 @@ class NetworkInference(BaseConfig):
'_eval.pt'))
def report_name(self, state_file):
- return pathlib.Path(str(state_file).replace(state_file.suffix,
- '_cr.tex'))
+ return str(state_file).replace(state_file.suffix, '_cr.tex')
def evaluate(self):
"""Evaluate the models on a defined dataset.
@@ -2664,7 +2663,8 @@ class NetworkInference(BaseConfig):
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))
+ report_name = self.report_path.joinpath(
+ self.report_name(state.name))
LOGGER.info('Calculating classification report: {}'
.format(report_name))
diff --git a/pysegcnn/core/utils.py b/pysegcnn/core/utils.py
index 65419ad..4384259 100644
--- a/pysegcnn/core/utils.py
+++ b/pysegcnn/core/utils.py
@@ -2655,7 +2655,7 @@ def report2latex(classification_report, filename=None):
if filename is not None:
filename = pathlib.Path(filename)
if not filename.exists():
- filename.parent().mkdir(exist_ok=True, parents=True)
+ filename.parent.mkdir(exist_ok=True, parents=True)
# export to latex
df.to_latex(buf=str(filename))
--
GitLab