Как вы можете уменьшить параметр ntree = 500 по умолчанию, переданный RF из каретки?

Question

Я считаю, что метод «rf» (randomForest) в caret устанавливает количество деревьев по умолчанию, равное 500. К сожалению, это приводит к тому, что временная сложность вырастает из-под контроля для больших наборов данных. Есть ли быстрый способ уменьшить количество деревьев без создания настраиваемого метода? Я знаю, что единственным настраиваемым параметром для rf является mtry.

Просто уточнить: я не хочу настраивать количество деревьев. Я просто хочу исправить его до более низкого значения, чтобы я мог запустить rf за разумное время.

Answer 1

Вы можете задать параметр ntree при вызове поезд, как так:

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

Answer 2

Одно из предложений было бы использовать библиотеку randomForest. Я всегда обнаружил, что один более простой в использовании, чем один в caret, и у него есть параметр для установки количества деревьев.