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

Question

如何在我的分析数据集中创建一个变量折叠，该数据集从createFolds交叉验证函数在caret中分配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.
? 

Answer 1

您可以使用createFolds来实现这一点，它有一个参数list，默认为TRUE，给出了每个折叠中的指标列表。如果将其更改为FALSE，则该函数将返回分配给每一行的折叠向量。

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

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