scikit-tree
scikit-tree copied to clipboard
neurodata
ONNX support
Does treeple support ONNX export? It would be great as ONNX is the common format for all sorts of models and can be deployed to any environment (e.g. scikit-learn, lightgbm, xgboost etc all support ONNX export).
No ONNX is not supported yet, but the package supports pickling in the same way scikit-learn does. I'm not familiar with how to get ONNX support working.
Are there any technical features required for that?
Thanks for the reply. skl2onnx library supports a lot of base models from scikit-learn. If treeple models can use similar classes, it should be straightforward.
Note that ONNX is quite different than pickling. It is safe, it provides inference optimizations, it is framework, programming language or environment agnostic.
Just to understand a bit better, what is your use-case for using ONNX instead of the actual pickled model?
Would you be interested in exploring how to get treeple models working with ONNX?
Because pickle is a python-specific format and one can not serve pickle in frontend (Typescript or javascript) or embed in some C++ application etc.
I may have a look during spring if I have time.