From 51b20fcddc4a339752b711c31795cbba68f799c1 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Fri, 5 Feb 2021 15:56:24 +0100
Subject: [PATCH] Improved logging for inference.

---
 pysegcnn/core/trainer.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 4fc4f2f..521ce56 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
-- 
GitLab