Изменить параметры по умолчанию для вычисления 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-репозитории пакета. Макс Кун, сопровождающий, довольно отзывчив, и у вас есть все шансы увидеть, как это будет реализовано в следующей версии.