Open bowerth opened 7 years ago
Sinew the book was written, an extra tuning parameter was added to the model code. You can use
gbmGrid <- expand.grid(interaction.depth=seq(1,6,by=1),
n.trees=c(25,50,100,200),
shrinkage=c(0.01,0.05,0.1),
n.minobsinnode = 10)
https://topepo.github.io/caret/model-training-and-tuning.html#basic-parameter-tuning
For a gradient boosting machine (GBM) model, there are three main tuning parameters: