-
Notifications
You must be signed in to change notification settings - Fork 402
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Multi-label ROCs #103
Labels
Comments
FlorianMF
added
enhancement
New feature or request
help wanted
Extra attention is needed
labels
Mar 17, 2021
Hi! thanks for your contribution!, great first issue! |
a possible implementation would be :
|
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
🚀 Feature
Similary to issue #100 it would be nice to be able to make
roc
work with multi-label inputs.Motivation & Pitch
auc
and hence_auroc_compute
do work with multi-label inputs and return a AUROC value for each label/class by iterating overrange(num_classes)
when passingaverage=None
._roc_compute
and henceroc
differentiate only between binary and multi-class (by checking ifnum_classes == 1
)I would expect
_roc_update
to similarly return a mode using_input_format_classification(preds, target)
and return a list of [fpr, tpr, threshold] of length=num_classes.The easiest would be the format
[[fpr, tpr, thres]]*5
The text was updated successfully, but these errors were encountered: