(y_pred: Tensor, y_true: Tensor, **kwargs)
| 75 | |
| 76 | |
| 77 | def roc_auc(y_pred: Tensor, y_true: Tensor, **kwargs): |
| 78 | # ROC-AUC calcualation |
| 79 | # Compute ROC curve and ROC area for each class |
| 80 | fpr = dict() |
| 81 | tpr = dict() |
| 82 | roc_auc = dict() |
| 83 | |
| 84 | y_true = y_true.detach().cpu().numpy() |
| 85 | y_pred = y_pred.detach().cpu().numpy() |
| 86 | |
| 87 | # Compute micro-average ROC curve and ROC area |
| 88 | fpr["micro"], tpr["micro"], _ = roc_curve(y_true.ravel(), y_pred.ravel()) |
| 89 | roc_auc["micro"] = auc(fpr["micro"], tpr["micro"]) |
| 90 | |
| 91 | return roc_auc["micro"] |
| 92 | |
| 93 | |
| 94 | def Hamming_loss( |