fix duplicate mask in f1_score (#145)

Summary: Pull Request resolved: #145 Reviewed By: ananthsub Differential Revision: D45188253 fbshipit-source-id: 6b29497cb849af78f80ea8aefdf7585a9476ac8a
pytorch · Apr 21, 2023 · d53fa32 · d53fa32
1 parent 30c6954
commit d53fa32
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/torcheval/metrics/functional/classification/f1_score.py b/torcheval/metrics/functional/classification/f1_score.py
@@ -227,8 +227,7 @@ def _f1_score_compute(
     elif average == "macro":
         return f1.mean()
     elif average == "weighted":
-        # pyre-fixme[61]: `mask` is undefined, or not always defined.
-        return (f1 * (num_label[mask] / num_label.sum())).sum()
+        return (f1 * (num_label / num_label.sum())).sum()
     else:  # average is None
         return f1