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

Implement ONNX export #23

Open
ogrisel opened this issue Nov 1, 2018 · 1 comment
Open

Implement ONNX export #23

ogrisel opened this issue Nov 1, 2018 · 1 comment

Comments

@ogrisel
Copy link
Owner

ogrisel commented Nov 1, 2018

https://github.com/onnx/onnxmltools has exports for lightgbm and scikit-learn decision trees.

It would be great to add onnx exports built-in pygbm, for instance in a new pygbm.onnx module (that is not imported by default so as to make the dependency on the onnx package optional).

This would also make it possible to improve the checks in test_compare_lightgbm so as to check that the structure of the generated trees are the same.

@ogrisel
Copy link
Owner Author

ogrisel commented Nov 1, 2018

Ping @xadupre if you are bored with scikit-learn :)

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

No branches or pull requests

1 participant