add none and unid

TheCacophonyProject · Oct 1, 2024 · 70a4ff8 · 70a4ff8
1 parent 2d1eb3f
commit 70a4ff8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/modelevaluate.py b/src/modelevaluate.py
@@ -242,6 +242,10 @@ def metadata_confusion(dir, confusion_file, after_date=None, model_metadata=None
         for k, v in remapped_labels.items():
             if v == "land-bird":
                 remapped_labels[k] = "bird"
+        if "None" not in labels:
+            labels.append("None")
+        if "unidentified" not in labels:
+            labels.append("unidentified")
     else:
         labels = [
             "bird",