import torch from torchmetrics.classification import MulticlassAUROC metric = MulticlassAUROC(num_classes=3) values = [ ] for _ in range(10): values.append(metric(torch.randn(20, 3), torch.randint(3, (20,)))) fig_, ax_ = metric.plot(values)