tidylearn (version 0.1.0)

tl_fit_boost: Fit a gradient boosting model

Description

Fit a gradient boosting model

Usage

tl_fit_boost(
  data,
  formula,
  is_classification = FALSE,
  n.trees = 100,
  interaction.depth = 3,
  shrinkage = 0.1,
  n.minobsinnode = 10,
  cv.folds = 0,
  ...
)

Value

A fitted gradient boosting model

Arguments

data

A data frame containing the training data

formula

A formula specifying the model

is_classification

Logical indicating if this is a classification problem

n.trees

Number of trees (default: 100)

interaction.depth

Depth of interactions (default: 3)

shrinkage

Learning rate (default: 0.1)

n.minobsinnode

Minimum number of observations in terminal nodes (default: 10)

cv.folds

Number of cross-validation folds (default: 0, no CV)

...

Additional arguments to pass to gbm()