torchmetrics icon indicating copy to clipboard operation
torchmetrics copied to clipboard

Multiple threshold for Multilabel metrics

Open ding3820 opened this issue 8 months ago • 0 comments

🚀 Feature

Allow multiple thresholds for every multilabel metrics

Motivation

I notice that for every multilabel metrics, threshold only allows a float. In some cases, we would like to use different threshold for different class.

Pitch

For example: mla = MultilabelAccuracy(threshold=[0.3, 0.7, 0.55], num_labels=3, average=None)

Alternatives

Additional context

ding3820 avatar Jun 28 '24 09:06 ding3820