在插入符号中为cv创建分层折叠的方法就是这样
library(caret)
library(data.table)
train_dat <- data.table(group = c(rep("group1",10), rep("group2",5)), x1 = rnorm(15), x2 = rnorm(15), label = factor(c(rep("treatment",15), rep("control",15))))
folds <- createFolds(train_dat[, group], k = 5)
fitCtrl <- trainControl(method = "cv", index = folds, classProbs = T, summaryFunction = twoClassSummary)
train(label~., data = train_dat[, !c("group"), with = F], trControl = fitCtrl, method = "xgbTree", metric = "ROC")
为了 balancer group1和group2,折叠索引的创建基于“group”变量 .
但是,有没有办法在插入符号中重复创建文件?因此,我可以对重复进行 balancer 分割 . 我应该组合几个createFolds并运行trainControl吗?
trControl = trainControl(method = "cv", index = many_repeated_folds)
谢谢!
1 回答
createMultiFolds
可能是你感兴趣的 .