search

cog-imperial / OMLT

Represent trained machine learning models as Pyomo optimization formulations
Other
281 stars 59 forks source link

LightGBM Linear Tree #162

Closed mu-wang closed 3 months ago

mu-wang commented 3 months ago

LightGBM offers linear tree implementation: https://lightgbm.readthedocs.io/en/latest/Parameters.html#dataset-parameters Will OMLT work for the onnx model exported from LightGBM linear tree?

rmisener commented 3 months ago

No, unfortunately not, the linear tree IO is different than the other models in OMLT. Take a look at the Jupyter notebook to see what input/output approaches work with linear trees. We're working on making the tree-related IO more sensible in OMLT (let us know if you want to help!), but it's not ready.