torcheval.metrics.functional.multiclass_auroc

torcheval.metrics.functional.multiclass_auroc(input: Tensor, target: Tensor, *, num_classes: int, average: str | None = 'macro') → Tensor

计算 AUROC，即 ROC 曲线下的面积，用于多类别分类。其类版本为 torcheval.metrics.MulticlassAUROC。

参数：

• input (Tensor) – 标签预测的张量 它应该是形状为 (n_sample, n_class) 的概率或 logits。

• target (Tensor) – 形状为 (n_samples, ) 的真值标签张量。

• num_classes (int) – 类别数量。

• average (str, 可选) –

  • 'macro' [默认]

    分别计算每个类别的指标，并返回它们的未加权平均值。

  • None:

    分别计算每个类别的指标，并返回每个类别的指标。

示例

>>> import torch
>>> from torcheval.metrics.functional import multiclass_auroc
>>> input = torch.tensor([[0.1, 0.1, 0.1, 0.1], [0.5, 0.5, 0.5, 0.5], [0.7, 0.7, 0.7, 0.7], [0.8, 0.8, 0.8, 0.8]])
>>> target = torch.tensor([0, 1, 2, 3])
>>> multiclass_auroc(input, target, num_classes=4)
0.5
>>> multiclass_auroc(input, target, num_classes=4, average=None)
tensor([0.0000, 0.3333, 0.6667, 1.0000])