我正在使用 "ROC" 作为度量,使用 rfe 函数( caret 包)计算 SVM-RFE 模型。据我所知,rfe 算法使用 roc 包的 pROC 函数及其预定义参数来优化 AUC 值。但是,我想将 direction 参数设置为 "<" 而不是 "auto" 因为在某些情况下,结果平均 AUC 是反向计算的(我的数据不是很好......)。这个问题在这里的答案中有解释:Difference in average AUC computation using ROCR and 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)

最佳答案

我查看了 caret 的源代码,目前似乎不太可能。参数不会传递到对 roc 函数的调用。

我建议在 the github repository of the package 上提交增强请求。 Max Kuhn 维护者非常敏感,你很有可能在 future 的版本中看到它的实现。

关于r - 更改 RFE 过程中 ROC 计算 (pROC) 的默认参数 (caret) - R,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/37410848/

10-11 16:49