diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py index 9501715eadb290c6c5d7708b105c6e42a208849e..f23a72aac4e41cea97e95e730b13b536a7ad66ae 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 65419ad0751db1ce650d6f037693564bad1a8bc3..43842590b9ce4112ebdfd8ba41e68450d28acb6f 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))