search

neurodata / treeple

Scikit-learn compatible decision trees beyond those offered in scikit-learn
https://treeple.ai
Other
61 stars 14 forks source link

Ensure compatability with hummingbird #316

Open adam2392 opened 3 weeks ago

adam2392 commented 3 weeks ago

Can we have an example of converting decision tree models to hummingbird for faster inference?

https://github.com/microsoft/hummingbird

In addition, an integration test would be great.

  1. Add hummingbird as a test dependency
  2. Fit every forest we have.
  3. Convert to hummingbird
  4. Verify inference on test set is numerically the same.

Perhaps Ryan or Vlad can take this?