Sets default values for average to micro throughout stat_scores. Fixes …

…Lightning-AI#2320, fixes, Lightning-AI#2047, resolves Lightning-AI#2280.
StefanoWoerner · Jan 23, 2024 · ede253c · ede253c
1 parent a68455a
commit ede253c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/torchmetrics/classification/stat_scores.py b/src/torchmetrics/classification/stat_scores.py
@@ -305,7 +305,7 @@ def __init__(
         self,
         num_classes: int,
         top_k: int = 1,
-        average: Optional[Literal["micro", "macro", "weighted", "none"]] = "macro",
+        average: Optional[Literal["micro", "macro", "weighted", "none"]] = "micro",
         multidim_average: Literal["global", "samplewise"] = "global",
         ignore_index: Optional[int] = None,
         validate_args: bool = True,
@@ -452,7 +452,7 @@ def __init__(
         self,
         num_labels: int,
         threshold: float = 0.5,
-        average: Optional[Literal["micro", "macro", "weighted", "none"]] = "macro",
+        average: Optional[Literal["micro", "macro", "weighted", "none"]] = "micro",
         multidim_average: Literal["global", "samplewise"] = "global",
         ignore_index: Optional[int] = None,
         validate_args: bool = True,