Dice metric cannot compute DICE score for each class

[NguyenAnhTien]

🐛 Bug

The class DICE (in the dice.py), at the line 176, the condition

if average not in ["micro", "macro", "samples"]

doest not work with the documents. This condition does not allow 'none' (None) value for the class DICE. Therefore, this class cannot produce DICE for each class.

[github-actions[bot]]

Hi! thanks for your contribution!, great first issue!

[Borda]

I guess you are referring to the master branch, right? Pls add a link to the code; that would be easier... :flamingo:

[NguyenAnhTien]

I figured out the problem in this version

https://github.com/Lightning-AI/torchmetrics/blob/v1.2.1/src/torchmetrics/classification/dice.py#L31-L288

[ManuelZ]

I have the same problem, I can't get a per-class Dice score when providing an average argument of None:

https://github.com/Lightning-AI/torchmetrics/blob/7e2ebff476d4faa56a5eb526732a880df40e7bb8/src/torchmetrics/classification/dice.py#L177

[asbjrnmunk]

Duplicate issue: https://github.com/Lightning-AI/torchmetrics/issues/1602