如何将用户指定的索引传递给插入符号trainControl？

Question

我正在使用需要大量预处理的数据集，为了防止过度拟合，我自己构建了交叉验证折叠。所以我有一个数据集，其中前k个记录属于fold 1（我还有一个fold_number列来指定记录所属的折叠）第二个k记录属于fold 2等。我想使用插入包并使用交叉验证调整参数，但我无法弄清楚如何将每个折叠的ID传递给trainControl（）函数，而不是让R交叉验证给我。有没有人有任何想法？

Answer 1

您正在寻找index的{​​{1}}和indexOut参数。这在文档trainControl

中明确定义

??trainControl

邮件列表here也解决了这个问题。