r - 如何减少从插入符号传递给 RF 的默认 ntree=500 参数？

Question

我相信 randomForest 中的“rf”( caret )方法将默认的树数设置为 500。不幸的是，这会导致时间复杂度对于较大的数据集失去控制。有没有什么快速的方法可以在不创建自定义方法的情况下减少树的数量？我知道 rf 的唯一可调参数是 mtry。

澄清一下:我不想调整树木的数量。我只是想将它修复为较低的值，以便我可以在合理的时间内运行 rf。

Answer 1

您可以在调用 train 时指定 ntree 参数，如下所示:

rf <- train(X, y, method="rf", preProcess=c("center","scale"), ntree=100, trControl=fitControl)