From 42c1f8dd0df25c7694d675a769e51273f10a8382 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Fri, 19 Feb 2021 16:13:53 +0100
Subject: [PATCH] Handle case when a class is not present in either ground
truth or prediction.
---
pysegcnn/core/trainer.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 6f34059..c8020f7 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -2717,7 +2717,8 @@ class NetworkInference(BaseConfig):
.format(report_name))
report = classification_report(
y_true, y_pred, target_names=self.class_names,
- output_dict=True, zero_division=1)
+ output_dict=True, zero_division=1,
+ labels=np.asarray(list(self.use_labels.keys())))
# store report in output dictionary
output['report'] = report2df(report, self.class_names)
--
GitLab