Как вычислить AUC под ROC в R (каретка, случайный лес, свм)

Question

Я использую метод случайного леса и метода опорных векторов в пакете каретки в R. Я хочу рассчитать AUC по ROC для обоих случаев;однако я не знаю, как это сделать в данном конкретном случае.Мой результат закодирован как 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"
)