如何通过插入符中的createFolds函数创建交叉验证折叠变量？

Question

如何在我的分析数据集中创建一个变量fold，该数据集从createFolds的{​​{1}}交叉验证函数中分配了值1：k？

例如，使用以下大学入学数据集：

caret

Answer 1

你可以使用createFolds执行此操作，它有一个参数list，默认为TRUE，列出每个折叠中的指标。如果将其更改为FALSE，则该函数将返回每行所分配的折叠向量。

folds <- createFolds(mydata$admit, k=5,list = FALSE)

有关详细信息，请参阅?createFolds。