Skip to content

4.0.0 Release

Compare
Choose a tag to compare
@hcho3 hcho3 released this 13 Jan 23:37
· 42 commits to mainline since this release
e878556

We are excited to announce Treelite 4.0.

What's new

  • Native support for multi-target models
  • Complete support for XGBoost 2.0, including vector-leaf trees (multi_strategy="multi_output_tree")
  • Greater variety of scikit-learn estimators, including HistGradientBoostingRegressor and HistGradientBoostingClassifier.
  • Vector base_scores, to support boosting from the average
  • Brand-new model builder API with built-in model validation
  • Ability to edit trees using field accessor API
  • We now provide a pre-built binary wheel targeting ARM64 (aarch64) architecture.

Bug fixes

  • Fix base_scores for sklearn GradientBoostingClassifier by @hcho3 in #536
  • Fix model concatenation by @hcho3 in #538
  • Don't fetch GTest if it's found by CMake by @trxcllnt in #540

Breaking change

  • The compiler (C codegen) module has been removed from Treelite. You may use TL2cgen for equivalent functionalities. See the migration guide.