This package provides the following bindings for parsnip package:

• the tree engine for decision_tree;
• the catboost engine for boost_tree - only available in catboost branch. See catboost;
• the lightGBM engine for boost_tree.

Note that the development of this package has shifted to the bonsai package. We suggest filing issues and/or pull requests there.

• tree package docs
• LightGBM docs
• Catboost docs

Not on CRAN yet.

remotes::install_github("curso-r/treesnip")

See catboost to use with catboost.

# decision_tree
model <- parsnip::decision_tree()
parsnip::set_engine(model, "tree")

# boost_tree
model <- parsnip::boost_tree(mtry = 1, trees = 50)

parsnip::set_engine(model, "catboost")
parsnip::set_engine(model, "lightgbm")

decision_tree()

boost_tree()

Originally treesnip had support for both lightgbm and catboost. Since catboost has no intent to make it to CRAN we removed the parsnip implementation from the main package. You can still use it from the catboost branch that we will keep up to date with the main branch.

The catboost branch can be installed with:

remotes::install_github("curso-r/treesnip@catboost")