allennlp icon indicating copy to clipboard operation
allennlp copied to clipboard

AUC for Multi-label Multi-class classification task

Open rvj07ai opened this issue 6 years ago • 3 comments

I am looking forward to have AUC for multi-label multi-class classification task but the AllenNLP library currently does not support it

Has anyone tried it earlier ?

rvj07ai avatar May 24 '19 04:05 rvj07ai

we have AUC for binary classification

https://github.com/allenai/allennlp/blob/master/allennlp/training/metrics/auc.py

but I don't know of anyone who's implemented it for multi-label / multi-class classification, sorry.

joelgrus avatar May 24 '19 04:05 joelgrus

Thanks @joelgrus - I have tried reusing the auc mentioned above to implement a custom multi-label / multi-class classification AUC . Working on it..

rvj07ai avatar May 24 '19 05:05 rvj07ai

@rvj07ai Are you still working on this feature? I'm thinking of starting on it.

mukhal avatar May 31 '19 15:05 mukhal