如何计算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"
)