我正在建立一个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 控件。

10-08 07:57