fixed typo in comment
This commit is contained in:
Shaunak Joshi 2023-03-31 16:51:59 -04:00 committed by GitHub
parent 78c3235eee
commit 85bbe15fcb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -57,7 +57,7 @@ class AUROCWithMWU(torchmetrics.Metric):
This AUROC implementation is well suited to (non-zero) low-CTR. In particular it will return
the correct AUROC even if the predicted probabilities are all close to 0.
Currently only support binary classification.
Currently only supports binary classification.
"""
def __init__(self, label_threshold: float = 0.5, raise_missing_class: bool = False, **kwargs):