-->

How to specify minbucket in caret train for?

2019-03-06 01:02发布

问题:

For CART model, caret seems to only provide tunning of complexity parameter. Is there way to tune other parameters such as minbucket?

回答1:

Arguments passed to the classification or regression routines are included in the dots parameter.

As you want to include minbucket, parameter control should be included inside train.

As example:

library("caret")

train(Kyphosis ~ Age + Number + Start, data = kyphosis, 
      method = "rpart",
      tuneGrid = data.frame(cp = c(0.01, 0.05)),
      control = rpart.control(minsplit = 10, minbucket = 5))

If you want to tune minbucket you will have to do it manually.



标签: r r-caret rpart