Skip to content

Commit

Permalink
Merge pull request #966 from brianhou0208/fix_metric_type
Browse files Browse the repository at this point in the history
Fix Metric typo
  • Loading branch information
adamjstewart authored Oct 27, 2024
2 parents 625ee55 + 7f39882 commit 783e403
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions segmentation_models_pytorch/metrics/functional.py
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,7 @@ def get_stats(
mode (str): One of ``'binary'`` | ``'multilabel'`` | ``'multiclass'``
ignore_index (Optional[int]): Label to ignore on for metric computation.
**Not** supproted for ``'binary'`` and ``'multilabel'`` modes. Defaults to None.
**Not** supported for ``'binary'`` and ``'multilabel'`` modes. Defaults to None.
threshold (Optional[float, List[float]]): Binarization threshold for
``output`` in case of ``'binary'`` or ``'multilabel'`` modes. Defaults to None.
num_classes (Optional[int]): Number of classes, necessary attribute
Expand Down Expand Up @@ -146,7 +146,7 @@ def get_stats(

if mode != "multiclass" and ignore_index is not None:
raise ValueError(
f"``ignore_index`` parameter is not supproted for '{mode}' mode"
f"``ignore_index`` parameter is not supported for '{mode}' mode"
)

if mode == "multiclass" and num_classes is None:
Expand Down

0 comments on commit 783e403

Please sign in to comment.