Skip to content
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

add support for quantile objective for LGBM models #503

Merged
merged 2 commits into from
Oct 4, 2021

Conversation

muyajil
Copy link
Contributor

@muyajil muyajil commented Sep 23, 2021

This PR adds support to directly convert LGBM models that use the objective 'quantile'.
There is no real difference in handling the output of a quantile model versus a regression model, since the quantile model is just a regression tuned to output a certain quantile, therefore it can be handled the same.

Signed-off-by: Mohammed Ajil <mohammed@ajil.ch>
@janjagusch
Copy link
Contributor

thanks for this PR!

i'm not a maintainer of this repo but i implemented a bit of stuff related to the LGBM objectives previously. i would suggest to add a test case here to make sure that your changes are working as expected. 🙂

Signed-off-by: Mohammed Ajil <mohammed@ajil.ch>
@xadupre xadupre merged commit f089575 into onnx:master Oct 4, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants