File tree 1 file changed +2
-2
lines changed
segmentation_models_pytorch/metrics
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -96,7 +96,7 @@ def get_stats(
96
96
97
97
mode (str): One of ``'binary'`` | ``'multilabel'`` | ``'multiclass'``
98
98
ignore_index (Optional[int]): Label to ignore on for metric computation.
99
- **Not** supproted for ``'binary'`` and ``'multilabel'`` modes. Defaults to None.
99
+ **Not** supported for ``'binary'`` and ``'multilabel'`` modes. Defaults to None.
100
100
threshold (Optional[float, List[float]]): Binarization threshold for
101
101
``output`` in case of ``'binary'`` or ``'multilabel'`` modes. Defaults to None.
102
102
num_classes (Optional[int]): Number of classes, necessary attribute
@@ -146,7 +146,7 @@ def get_stats(
146
146
147
147
if mode != "multiclass" and ignore_index is not None :
148
148
raise ValueError (
149
- f"``ignore_index`` parameter is not supproted for '{ mode } ' mode"
149
+ f"``ignore_index`` parameter is not supported for '{ mode } ' mode"
150
150
)
151
151
152
152
if mode == "multiclass" and num_classes is None :
You can’t perform that action at this time.
0 commit comments