AUC & ROC for MultiClass Classification CNN

[muellerzr (Zachary Mueller)]

Right here:

AUC for multiclass classification

Adapted Using AUC as metric in fastai for multiclass from sklearn.metrics import roc_auc_score def auroc_score(input, target): input, target = input.cpu().numpy(), target.cpu().numpy() return roc_auc_score(target, input) @dataclass class AUC(Callback): "Computes the auc for multi-class classification" clas:int=1 def on_epoch_begin(self, **kwargs): self.output, self.target = [], [] def on_batch_end(self, last_output, last_target, train, **kwargs): if …