v1.2.0
What's New
We continuously update our library by making improvements and adding new features. If you use or like our library, please star⭐ this repo. Thank you!
Major Improvements
- In this version,
AUCMLoss
can automatically computeimratio
without requiring this input from users. - Renamed
gamma
toepoch_decay
forPESG
andPDSCA
optimizers, i.e.,epoch_decay
=1/gamma
- Reimplemented
ImbalancedDataGenerator
for constructing imbalanced dataset for benchmarking. Tutorial is available here. - Improved implementations of
APLoss
by removing some redundant computations. - Merged
SOAP_ADAM
andSOAP_SGD
optimizers into one optimizerSOAP
. Tutorial is provided here. - Removed dependency of
TensorFlow
and now LibAUC only requiresPyTorch
installed . - Updated existing tutorials to match the new version of LibAUC. Tutorials are available here.
New Features
- Introduced
DualSampler
,TriSampler
for sampling data that best fit the x-risk optimization to balance inner and outer estimation error. - Introduced
CompositionAUCLoss
andPDSCA
optimizer. Tutorial is provided here. - Introduced
SogCLR
withDynamic Contrastive Loss
for training Self-Supervised Learning models using small batch size. Tutorial and code are provided here. - Introduced
NDCG_Loss
andSONG
optimizer for optimizing NDCG. Tutorials are provided here. - Introduced
pAUCLoss
with three optimizers:SOPA
,SOPAs
,SOTAs
for optimizing Partial AUROC. Tutorials are provided here. - Added three evaluation functions:
auc_roc_score
(binary/multi-task),auc_prc_score
(binary/multi-task) andpauc_roc_score
(binary).
Feedback
- If you have any feedback/suggestions, please contact us @ Zhuoning Yuan [yzhuoning@gmail.com] and Tianbao Yang [tianbao-yang@uiowa.edu].