Fix docstring typos for MulticlassAveragePrecision, `MultilabelAver…

…agePrecision` (#1820)
Lightning-AI · Jun 3, 2023 · 5a01f34 · 5a01f34
1 parent c03a716
commit 5a01f34
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/torchmetrics/classification/average_precision.py b/src/torchmetrics/classification/average_precision.py
@@ -158,7 +158,7 @@ def plot(
 
 
 class MulticlassAveragePrecision(MulticlassPrecisionRecallCurve):
-    r"""Compute the average precision (AP) score for binary tasks.
+    r"""Compute the average precision (AP) score for multiclass tasks.
 
     The AP score summarizes a precision-recall curve as an weighted mean of precisions at each threshold, with the
     difference in recall from the previous threshold as weight:
@@ -307,7 +307,7 @@ def plot(
 
 
 class MultilabelAveragePrecision(MultilabelPrecisionRecallCurve):
-    r"""Compute the average precision (AP) score for binary tasks.
+    r"""Compute the average precision (AP) score for multilabel tasks.
 
     The AP score summarizes a precision-recall curve as an weighted mean of precisions at each threshold, with the
     difference in recall from the previous threshold as weight: