diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py index 7df3bca93afef88a9525327ff351179973e99b94..9501715eadb290c6c5d7708b105c6e42a208849e 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: {}'