add ignore_index to default_metric_classwise

FrenchKrab · May 11, 2023 · 5f51b50 · 5f51b50
1 parent 05638c7
commit 5f51b50
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pyannote/audio/tasks/segmentation/multilabel.py b/pyannote/audio/tasks/segmentation/multilabel.py
@@ -343,9 +343,9 @@ def default_metric_classwise(
         self,
     ) -> Union[Metric, Sequence[Metric], Dict[str, Metric]]:
         return [
-            F1Score(task="binary"),
-            Precision(task="binary"),
-            Recall(task="binary"),
+            F1Score(task="binary", ignore_index=-1),
+            Precision(task="binary", ignore_index=-1),
+            Recall(task="binary", ignore_index=-1),
         ]
 
     @cached_property