How to specify minbucket in caret train for?

前端 未结 1 1074
粉色の甜心
粉色の甜心 2021-01-25 17:12

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

相关标签:
1条回答
  • 2021-01-25 18:17

    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.

    0 讨论(0)
提交回复
热议问题