如何在R中的ROC下計算AUC（插入符號，隨機森林，svm）

Question

我在R中的插入符號包中使用隨機森林和支持向量機方法。我想在兩種情況下都在ROC下計算AUC。 但是，我不知道在這種特殊情況下該怎么做。 我的結果編碼為0和1。這是我正在使用的代碼示例：

set.seed(123)
cvCtrl <- trainControl(method = "cv", number = 10)
rf_moded<-train(readm30~.,data=train,method="rf", trControl=cvCtrl)

Answer 1

您要使用ROC訓練模型嗎？ 然后，您需要以下內容：

對於trainControl ：

control <- trainControl(method = 'cv', number = 10, 
  savePredictions = 'final', classProbs = TRUE, summaryFunction = twoClassSummary)

在train ：

train(
  outcome ~ .,
  data = data,
  method = method,
  trControl = control,
  metric = "ROC"
)