We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 30c6954 commit d53fa32Copy full SHA for d53fa32
torcheval/metrics/functional/classification/f1_score.py
@@ -227,8 +227,7 @@ def _f1_score_compute(
227
elif average == "macro":
228
return f1.mean()
229
elif average == "weighted":
230
- # pyre-fixme[61]: `mask` is undefined, or not always defined.
231
- return (f1 * (num_label[mask] / num_label.sum())).sum()
+ return (f1 * (num_label / num_label.sum())).sum()
232
else: # average is None
233
return f1
234
0 commit comments