Получите прогнозы от каждого дерева в модели случайного леса в R (используя функцию поезда в обучении и прогнозирование в прогнозировании) - PullRequest
1 голос
/ 30 сентября 2019

Я использую функцию train при обучении модели случайного леса:

fitControl = trainControl(method="oob")
tuneGrid = expand.grid(.mtry=c(15))    

rfmod = train(p ~   
            x +  
            y +   
            z,             
          method="rf",
          data=train_df,
          tuneGrid=tuneGrid,
          trControl=fitControl,
          importance=TRUE,
          allowParallel=TRUE)

Это упрощенный код, показывающий структуру моей модели, и я использую данные обучения train_df.

И я хочу получить прогноз от каждого дерева. Я искал как-то и пробовал это:

preds <- predict(rfmod, newdata = test_df[1], predict.all = TRUE)

Я просто использовал первый ряд моих тестовых данных test_df. После этого, когда я проверяю preds, он по-прежнему дает мне только одно значение прогнозирования, в то время как я хочу получить прогноз для всех деревьев.

Как я могу сделать это, чтобы получить все прогнозы от всех деревьев?

Заранее спасибо !!

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...