Я использую функцию 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
, он по-прежнему дает мне только одно значение прогнозирования, в то время как я хочу получить прогноз для всех деревьев.
Как я могу сделать это, чтобы получить все прогнозы от всех деревьев?
Заранее спасибо !!