-
Notifications
You must be signed in to change notification settings - Fork 3.8k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
LightGBM Earth Mover's Distance #1256
Comments
Could you specify more precisely what you have in mind? Even computing the EMD requires solving a linear program if I'm not mistaken. |
@vruvora , what's situation/scenario do you want to use Wasserstein/EMD? considering it's normally seen in deep learning scenario. |
I want to train the LightGBM to output a distribution and over a fixed ordered support. Cross-entropy would not account for this ordering e.g. the loss should be aggregated lower closer the index is to actual index. |
Closed in favor of being in #2302. We decided to keep all feature requests in one place. Welcome to contribute this feature! Please re-open this issue (or post a comment if you are not a topic starter) if you are actively working on implementing this feature. |
Thanks for designing such an awesome library. I was curious if you have plans to incorporate Earth Mover's Distance into LightGBM so we can optimize over distributions.
The text was updated successfully, but these errors were encountered: