Optimization-AI / LibAUC

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

ImbalancedDataGenerator() ISSUE #23

Closed rohan1561 closed 2 years ago

rohan1561 commented 2 years ago

Hello, nice work! I was just wondering why you convert every label <= split index into 0 and all others into 1 (lines 62 and 63 in utils/generator.py)? That makes the task binary but it just pools several classes into two. Have I misunderstood something? Why would one want to do this?

yzhuoning commented 2 years ago

Hello, this is one way that you could construct the imbalanced dataset with customized imbalanced ratio (e.g., pos/all). You can also try other methods to construct imbalanced dataset (e.g., predicting one class vs the rest classes in CIFAR10) or directly use the naturally imbalanced datasets.