我正在建立一个CART模型,并且试图调整rpart的2个参数-CP和Maxdepth。虽然插入符号包一次只能对一个参数起作用,但是当同时使用两个参数时,它总是抛出错误,我无法弄清为什么
library(caret)
data(iris)
tc <- trainControl("cv",10)
rpart.grid <- expand.grid(cp=seq(0,0.1,0.01), minsplit=c(10,20))
train(Petal.Width ~ Petal.Length + Sepal.Width + Sepal.Length, data=iris, method="rpart",
trControl=tc, tuneGrid=rpart.grid)
我收到以下错误:
Error in train.default(x, y, weights = w, ...) :
The tuning parameter grid should have columns cp
最佳答案
方法“rpart”只能调整cp,方法“rpart2”用于maxdepth。没有对 minsplit 或任何其他 rpart 控件进行调整。如果您想调整不同的选项,您可以编写一个自定义模型来考虑这一点。
单击 here 以获取有关如何执行此操作的更多信息。另请阅读 this answer 了解如何在 train 函数中使用 rpart 控件。