thesps / conifer

Fast inference of Boosted Decision Trees in FPGAs
Apache License 2.0
48 stars 27 forks source link

Add support for Yggdrasil / TensorFlow Decision Forests models #34

Closed achoum closed 1 year ago

achoum commented 1 year ago

Following this change, a subset of Yggdrasil / TensorFlow Decision Forests models can be imported to and compiled by Conifer. This change supports Classification (binary and multi-class) Gradient Boosted Trees with feature>=threshold conditions.

thesps commented 1 year ago

Thanks for this, it's super nice to have support for these models. Merging!