Jak zmniejszyć domyślny parametr ntree = 500 przekazany do RF z karetki?

Question

Wierzę, że metoda "rf"() w caret ustawia domyślną liczbę drzew na 500. Niestety, powoduje to, że złożoność czasu wzrasta poza kontrolą w przypadku większych zestawów danych. Czy istnieje jakiś szybki sposób na zmniejszenie liczby drzew bez tworzenia niestandardowej metody? Wiem, że jedynym dostrajalnym parametrem dla rf jest mtry.

Po prostu wyjaśnić: Nie zamierzam stroić na liczbę drzew. Po prostu chcę go naprawić na niższą wartość, aby móc uruchomić rf w rozsądnym czasie.

Answer 1

można określić parametr ntree kiedy zadzwonić pociąg tak:

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

Answer 2

Jedną z sugestii jest użycie biblioteki randomForest. Zawsze uważałem, że jest prostszy w użyciu niż ten w caret i ma parametr do ustawienia liczby drzew.