PyTorch Lightning AUROC value for multi-class seems to be completely off compared to sklearn (using it wrong)?

So far this is only a binary implementation.

Basically we have a multiclass auc implementation here and a multiclass roc calculation here which you can then combine to a multiclass auric as it has been done here

1 Like