search

dmlc / treelite

Universal model exchange and serialization format for decision tree forests
https://treelite.readthedocs.io/en/latest/
Apache License 2.0
723 stars 98 forks source link

Export model back to scikit-learn #558

Closed Mohit1908 closed 1 month ago

Mohit1908 commented 3 months ago

Method to export model back to sklearn. This would help in analysing/porting models better.

hcho3 commented 1 month ago

This is difficult, because the Treelite model spec has some features that are not supported by scikit-learn (e.g. ability to set comparison ops in test nodes).