From 25b2715c443b44a243aa439020ced69a5b2eb56e Mon Sep 17 00:00:00 2001 From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu> Date: Sat, 20 Feb 2021 10:01:32 +0100 Subject: [PATCH] Fixed typo that messed up aggregated confusion matrix. --- 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 792ebe2..e663a33 100644 --- a/pysegcnn/core/trainer.py +++ b/pysegcnn/core/trainer.py @@ -2784,7 +2784,7 @@ class NetworkInference(BaseConfig): # check if for some scenes micro average is calculated rather # than accuracy due to missing class labels labels = list(report.index.drop(['macro avg', 'weighted avg', - 'micro_avg', 'accuracy'])) + 'micro avg', 'accuracy'])) except KeyError: # micro average is not calculated for any scene: each scene # contains all the classes -- GitLab