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]

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... 🦩

[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:

torchmetrics/src/torchmetrics/classification/dice.py

Line 177 in 7e2ebff

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

[asbjrnmunk]

Duplicate issue: #1602