r - 插入符中的 rfeControl 函数是否会创建分层折叠？

Question

我想按照 rfe 的方法对我的随机福雷斯特模型进行特征选择插入符包的。由于我的数据集仅包含大约 100 个标记样本，并且高度不平衡(反射(reflect)了现实生活的平衡)，因此我需要/想要进行分层交叉验证。但是，我没有找到任何有关分层交叉验证的 rfeControl 函数的文档。

有人知道如果我使用rfeControl函数是否会创建分层折叠

ctrl <- rfeControl(functions = rfFuncs,
               method = "cv",
               verbose = FALSE)

Answer 1

对于method ="cv"，rfe()应该使用createFolds()来创建你的折叠，并且这些将基于平衡在你的输出变量上。您可以查看 ?createFolds 了解如何实现的详细信息。