Изменить параметры по умолчанию для вычисления ROC (pROC) в процессе RFE (каретка) - R

Я вычисляю модель SVM-RFE, используя "ROC" в качестве метрики, с rfe функция (caret пакет). Насколько мне известно, rfe алгоритм оптимизирует значения AUC, используя roc функция pROC пакет, с его предопределенными аргументами. Тем не менее, я хотел бы установить direction аргумент "<" вместо "auto" потому что в некоторых случаях результирующий средний AUC рассчитывается в обратном порядке (мои данные не очень хорошие...). Эта проблема объясняется в ответе здесь: Разница в среднем вычислении AUC с использованием ROCR и pROC (R)

Как я могу изменить значение этого аргумента по умолчанию roc функция в rfe вычисление?

Я попробовал этот простой вариант, но он не работает:

svmRFE_NG3 <- rfe(x = TAll[,2:50],
               y = TAll[,1],
               sizes = seq(1,42),
               metric = "ROC",
               levels = c("BREAST","LUNG"),
               direction = "<",
               rfeControl = FSctrl,
               ## Options to train()
               method = "svmLinear",
               tuneLength = 10,
               preProc = c("center", "scale"),
               ## Inner resampling process
               trControl = TRctrl)

1 ответ

Решение

Я взглянул на исходный код карета, и в настоящий момент это не представляется возможным. Аргументы не передаются в призыв к roc функция.

Я бы предложил отправить запрос на улучшение в github-репозитории пакета. Макс Кун, сопровождающий, довольно отзывчив, и у вас есть все шансы увидеть, как это будет реализовано в следующей версии.

Другие вопросы по тегам