插入符号中的rfeControl功能是否会产生分层折叠？

Question

我希望按照插入符号包rfe的方法选择我的随机forrest模型。由于我的数据集仅包含约100个标记样本，并且由于它非常不平衡（这反映了实际生活平衡），我需要/想要进行分层交叉验证。但是，我没有找到关于分层交叉验证的rfeControl函数的任何文档。

如果我使用

，是否有人知道rfeControl函数是否确实创建了分层折叠

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

Answer 1

method ="cv"，rfe()应使用createFolds()创建折叠，这些将根据您的输出变量进行平衡。 您可以查看?createFolds，了解有关如何实施此功能的详细信息。