false pos and neg items now correctly updated

AstraZeneca · Oct 24, 2024 · 64c9ffb · 64c9ffb
1 parent e26487c
commit 64c9ffb
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/kazu/training/train_multilabel_ner.py b/kazu/training/train_multilabel_ner.py
@@ -398,15 +398,17 @@ def calculate_metrics(
             all_results[f"{clazz}_precision"] = result.precision
             all_results[f"{clazz}_recall"] = result.recall
             all_results[f"{clazz}_support"] = support
+            all_results["false_positives"] = {}
+            all_results["false_negatives"] = {}
 
             false_positives: defaultdict[str, dict[str, int]] = defaultdict(dict)
             for match, count in result.fp_info:
                 false_positives[clazz][match] = count
-            all_results["false_positives"] = dict(false_positives)
+            all_results["false_positives"].update(dict(false_positives))
             false_negatives: defaultdict[str, dict[str, int]] = defaultdict(dict)
             for match, count in result.fn_info:
                 false_negatives[clazz][match] = count
-            all_results["false_negatives"] = dict(false_negatives)
+            all_results["false_negatives"].update(dict(false_negatives))
         label_set = set(label_list)
         label_set.remove(ENTITY_OUTSIDE_SYMBOL)
         if len(ner_results) != len(label_set):