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

How to create a cross-validation fold variable via createFolds function in caret?

如何在我的分析数据集中创建一个变量 fold,该变量从 caret 中的 createFolds 交叉验证函数中分配了 1:k 的值?

例如使用以下大学招生数据集:

# Load data.
mydata <- read.csv("http://www.ats.ucla.edu/stat/data/binary.csv")
folds <- createFolds(mydata$admit, k=5)

# Create variable "mydata$fold" and assign values of 1:5 to mydata$fold.
? 

你可以用 createFolds 来做到这一点,它有一个参数 list 默认为 TRUE 给出每个折叠中的索引列表。如果将其更改为 FALSE 函数 returns 将每行分配给的折叠向量。

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

有关详细信息,请参阅 ?createFolds