Как использовать метрику ROC с RFE в каретке

Как использовать метрику ROC для ссылки? Я пробовал это ниже и получаю предупреждение, что «Точность» используется с намерением ROC.

rfFuncs$SummaryFunction <- twoClassSummary
ctrl_rfe <- rfeControl(method = "cv",
                   number = 5,
                   verbose = TRUE,
                   functions = rfFuncs,
                   allowParallel = F)
rf_rfe <- rfe(select(trainData, -target), 
          trainData$target, 
          sizes      = seq(1, ncol(trainData), 100), 
          metric='ROC',
          rfeControl = ctrl_rfe)

person Fred R.    schedule 08.09.2015    source источник


Ответы (1)


rfFuncs имеет summary, а не SummaryFunction.

Максимум

person topepo    schedule 09.09.2015