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

Does this method apply to linear regression models like elastic net? #11

Closed albert-ying closed 2 years ago

albert-ying commented 2 years ago

Hi, I'm wondering whether I can use this method as a preprocessing step for non-DNN models, namely simple linear regression or elastic net regression? If so, how should I adopt this method?

Thank you so much!

YyzHarry commented 2 years ago

Hi, thanks for your interest! Yes, for LDS, since it operates on the label space, it is independent of the model you use. You can refer to our tutorial for a quick starting, where you only need to replace the NN model with whatever regression model you like.

As for FDS, it might be less straightforward; but since it only needs feature statistics, if the algorithm can generate intermediate features for each input, it can also be adapted, depending on the actual method you use.

albert-ying commented 2 years ago

Thank you so much! I will close this issue for now!