在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后，我发现在randomForest()函数的参数中设置n=10或使用gbm时设置n.trees可能会有所帮助，但我对Caret包感兴趣。

非常感谢您的任何反馈。

谢谢

- owner

Answer 1

Caret的`train()`函数可以使用`randomForest()`函数来进行训练，只需在`train`调用时指定`method = "rf"`即可。

您只需要向`train`传递`ntree = 10`，该参数将会被传递到`randomForest()`中。

因此，您的调用应该是这样的：

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

- alex23lemm

---

感谢您的解释，我很感激。干杯！ - owner

Answer 2

对于任何使用随机森林ranger方法（即使在搜索时特别指定“ranger”），并且在此处着陆的人感兴趣，请使用num.trees参数。

num.trees = 20

- Doug Fir

Answer 3

我认为ntree是您正在寻找的参数。

- Maksim Gayduk