Replies: 1 comment
-
Some of the functions needed to create the optimization formulation of the tree ensemble with categorical vars are given in the following public branch: https://github.com/cog-imperial/OMLT/tree/cat_var_support |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Currently ONNX doesn't support categorical vars which limits the use of tree ensembles in OMLT. By directly interfacing with these libraries, we can add support for categorical vars.
Beta Was this translation helpful? Give feedback.
All reactions