Inconsistent default average
value in F1Score (and others) compared to MultiClassF1Score
#2047
Labels
average
value in F1Score (and others) compared to MultiClassF1Score
#2047
🐛 Bug
The default value for the
average
field in F1Score ismicro
:torchmetrics/src/torchmetrics/classification/f_beta.py
Line 1107 in fb84f75
However, the default in MultiClassF1Score is
macro
:torchmetrics/src/torchmetrics/classification/f_beta.py
Line 785 in fb84f75
When using the F1Score function, it seems as though that
micro
will be used by default, which is misleading.In practice, when using F1Score with
task='multiclass'
the default value ismacro
.*The same inconsistency exists in Precision and Recall (and maybe others).
To Reproduce
See F1Score / MultiClassF1Score function calls.
Expected behavior
Maybe delete the default value in the F1Score wrapper, as it is anyways overridden?
Environment
conda
,pip
, build from source): 0.11.4, condaThe text was updated successfully, but these errors were encountered: