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

时间:2019-10-23 15:43:35

标签: r random-forest r-caret

我正在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
)

]

0 个答案:

没有答案