diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py index 4fc4f2f0d5ab65267e297e40d7f53b379ac057f6..521ce5684f6632058aa9a19384df0a13c64a0a25 100644 --- a/pysegcnn/core/trainer.py +++ b/pysegcnn/core/trainer.py @@ -2661,7 +2661,7 @@ class NetworkInference(BaseConfig): # plot confusion matrix plot_confusion_matrix( conf_mat, self.source_labels, state_file=state, - subset=self.domain + '_' + self.trg_ds.name, + subset=self.trg_ds.name, outpath=self.perfmc_path) # save model predictions to file