Become an expert in R — Interactive courses, Cheat Sheets, certificates and more!
Get Started for Free

GBMModel

Generalized Boosted Regression Model


Description

Fits generalized boosted regression models.

Usage

GBMModel(
  distribution = NULL,
  n.trees = 100,
  interaction.depth = 1,
  n.minobsinnode = 10,
  shrinkage = 0.1,
  bag.fraction = 0.5
)

Arguments

distribution

optional character string specifying the name of the distribution to use or list with a component name specifying the distribution and any additional parameters needed. Set automatically according to the class type of the response variable.

n.trees

total number of trees to fit.

interaction.depth

maximum depth of variable interactions.

n.minobsinnode

minimum number of observations in the trees terminal nodes.

shrinkage

shrinkage parameter applied to each tree in the expansion.

bag.fraction

fraction of the training set observations randomly selected to propose the next tree in the expansion.

Details

Response Types:

factor, numeric, PoissonVariate, Surv

Automatic Tuning of Grid Parameters:

n.trees, interaction.depth, shrinkage*, n.minobsinnode*

* included only in randomly sampled grid points

Default values for the NULL arguments and further model details can be found in the source link below.

Value

MLModel class object.

See Also

Examples

## Requires prior installation of suggested package gbm to run

fit(Species ~ ., data = iris, model = GBMModel)

MachineShop

Machine Learning Models and Tools

v2.8.0
GPL-3
Authors
Brian J Smith [aut, cre]
Initial release
2021-04-16

We don't support your browser anymore

Please choose more modern alternatives, such as Google Chrome or Mozilla Firefox.