exclude unknown tag

TheCacophonyProject · Oct 1, 2024 · c6c6abb · c6c6abb
1 parent d7e3c10
commit c6c6abb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/modelevaluate.py b/src/modelevaluate.py
@@ -310,6 +310,9 @@ def metadata_confusion(dir, confusion_file, after_date=None, model_metadata=None
                     "Remapping %s to %s", human_tag, remapped_labels[human_tag]
                 )
                 human_tag = remapped_labels[human_tag]
+            if human_tag not in labels:
+                logging.info("Excluding %s", human_tag)
+
             ai_tags = []
             for tag in tags:
                 if tag.get("automatic") is True:
@@ -321,8 +324,6 @@ def metadata_confusion(dir, confusion_file, after_date=None, model_metadata=None
                         ai_tags.append(tag["what"])
 
             y_true.append(human_tag)
-            if human_tag not in labels:
-                labels.append(human_tag)
             if len(ai_tags) == 0:
                 y_pred.append("None")
             else: