search

YyzHarry / imbalanced-regression

[ICML 2021, Long Talk] Delving into Deep Imbalanced Regression
http://dir.csail.mit.edu
MIT License
819 stars 129 forks source link

bucket_num bucket_start #38

Closed wklix closed 11 months ago

wklix commented 1 year ago

Would you please tell me how to choose the best bucket_num and bucket_start parameters?The range of labels in my dataset is (-2, 16)

YyzHarry commented 11 months ago

Hi, thanks for your interest. The choice of bucket_num & bucket_start will depend on the (minimum) resolution you care about for your task. For example, in age prediction, the minimum interval we care about is 1 (year), so we chose bucket_num accordingly.