Update accuracy.py

Lightning-AI · Jan 8, 2025 · de8bd1e · de8bd1e
1 parent 8f510bb
commit de8bd1e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/torchmetrics/classification/accuracy.py b/src/torchmetrics/classification/accuracy.py
@@ -239,7 +239,7 @@ class MulticlassAccuracy(MulticlassStatScores):
         >>> preds = tensor([[[0, 2], [2, 0], [0, 1]], [[2, 2], [2, 1], [1, 0]]])
         >>> metric = MulticlassAccuracy(num_classes=3, multidim_average='samplewise')
         >>> metric(preds, target)
-        tensor([0.5000, 0.2778])
+        tensor([0.5000, 0.3333])
         >>> mca = MulticlassAccuracy(num_classes=3, multidim_average='samplewise', average=None)
         >>> mca(preds, target)
         tensor([[1.0000, 0.0000, 0.5000],