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

Question

我想按照 caret 包的 rfe 方法对我的随机 forrest 模型进行特征选择。由于我的数据集仅包含大约 100 个标记样本，并且高度不平衡（反映了现实生活中的平衡），因此我 need/want 进行分层交叉验证。但是，我没有找到任何有关分层交叉验证的 rfeControl 函数的文档。

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

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

Answer 1

与 method ="cv"、rfe() 应使用 createFolds() 来创建你的折叠，这些将根据你的输出变量进行平衡。 您可以查看 ?createFolds 以了解有关其实现方式的详细信息。