我正在使用 "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/