キャレットでrepeatedcvの階層化折り畳みを作成する方法は？

Question

キャレットでのCVのための成層折り目を作成する方法は、group1とgroup2のバランスをとるために、この

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") 

のようなものです、折り目インデックスの作成は、「グループ」の変数に基づいています。

しかし、キャレットにrepetcvのFoldsを作成する方法はありますか？ですから、私はrepeatedcvのバランスをとって分割することができます。いくつかのcreateFoldsを組み合わせてtrainControlを実行する必要がありますか？

trControl = trainControl(method = "cv", index = many_repeated_folds) 

ありがとう！

Answer 1

createMultiFoldsは、あなたが興味を持っているものをおそらくある。