bonsai provides bindings for additional tree-based model engines for use with the parsnip package.
This package is based off of the work done in the treesnip repository by Athos Damiani, Daniel Falbel, and Roel Hogervorst. bonsai is the official CRAN version of the package; new development will reside here.
You can install the most recent official release of bonsai with:
install.packages("bonsai")You can install the development version of bonsai from GitHub with:
# install.packages("pak")
pak::pak("tidymodels/bonsai")The bonsai package provides additional engines for the models in the following table:
| model | engine | mode |
|---|---|---|
| boost_tree | lightgbm | regression |
| boost_tree | lightgbm | classification |
| boost_tree | catboost | regression |
| boost_tree | catboost | classification |
| decision_tree | partykit | regression |
| decision_tree | partykit | classification |
| rand_forest | partykit | regression |
| rand_forest | partykit | classification |
| rand_forest | aorsf | classification |
| rand_forest | aorsf | regression |
Note that the catboost engine package isn’t available on CRAN, and should thus be installed according to their documentation.
Please note that the bonsai project is released with a Contributor Code of Conduct. By contributing to this project, you agree to abide by its terms.