r - 在 R ~ Caret 包中设置树的数量

Question

我目前想知道如何使用 Caret 包中的随机森林算法设置 10 棵树，希望能得到帮助:

下面是我的语法:

tr <- trainControl(method = "repeatedcv",number = 20)

fit<-train(y ~.,method="rf",data=example, trControl=tr)

关注 http://www.inside-r.org/packages/cran/randomForest/docs/randomForest

设置 n=10作为 randomForest() 中的参数或 n.trees如果使用 gbm可能只是有所帮助，但我对 Caret 感兴趣包裹。

任何反馈将不胜感激。

谢谢

Answer 1

插入符号 train()使用 randomForest()指定 method = "rf" 时的函数在 train称呼。

您只需要通过 ntree = 10至 train然后将传递给 randomForest() .

因此，您的电话将如下所示:
fit <- train(y ~., method="rf",data=example, trControl=tr, ntree = 10)