Optimization-AI / LibAUC

LibAUC: A Deep Learning Library for X-Risk Optimization
https://libauc.org/
MIT License
273 stars 37 forks source link

how to use pAUC loss on a validation dataset #22

Closed BoredGeo closed 1 year ago

BoredGeo commented 1 year ago

Hello, thank you for the nice package. I have a question - is it possible to use partial AUC loss on a validation dataset? I saw that the indices of the positive images are meaningful for loss calculation, and I was wondering if I can add the validation dataset images to the array without impacting the loss calculation (they will stay 0 during training) and getting a meaningful value during the validation step. Thanks.