r - R 中带插入符号的随机森林回归的 classProbs

Question

我正在 R 中执行随机森林回归，但希望得到类似于概率类的东西作为随机森林分类。过去，我运行过随机森林分类模型，并且能够获得预测的准确度。随机森林回归中是否有类似的概念？

对于以下代码：

fitControl <- trainControl(## 5-fold CV
  method = "cv",
  number = 5,
  search = 'grid',
  classProbs = TRUE,
  savePredictions = "final"
)
grid <- expand.grid(.mtry=seq(1,4,1))
set.seed(123)
rf_model <- train(iris$Sepal.length~.,
                  data = iris[-5],
                  method = "rf",
                  importance=TRUE,
                  trControl = fitControl
                  metric = "MAE",
                  tuneGrid=grid,
                  ntree=100
)

]

标签： rrandom-forestr-caret