设置 R ~ Caret 包中的树数

Question

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

下面是我的语法：

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

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

的研究

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

如有任何反馈，我们将不胜感激。

谢谢

Answer 1

我认为 ntree 是您要查找的参数

Answer 2

当您在 train 调用中指定 method = "rf" 时，Caret 的 train() 使用 randomForest() 函数。

您只需将 ntree = 10 传递给 train，后者将传递给 randomForest()。

因此，您的调用将如下所示：

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

Answer 3

对于任何在我的位置上使用随机 forrest 的游侠方法登陆这里的人感兴趣（Google 在我的搜索词中指定 "ranger" 时仍然将我定向到这里）使用 num.trees .

num.trees = 20

set number of trees in R ~ Caret package