插入包：列车功能的分层交叉验证-Java 学习之路

有没有办法在使用列车功能将模型拟合到大的不 balancer 数据集时执行分层交叉验证？我知道直接k折叠交叉验证是可能的，但我的类别非常不 balancer . 我已经看过关于这个主题的讨论，但没有真正明确的答案 .

提前致谢 .

1 回答

有一个名为'index'的参数，可以让用户指定索引进行交叉验证 .

folds <- 4
cvIndex <- createFolds(factor(training$Y), folds, returnTrain = T)
tc <- trainControl(index = cvIndex,
               method = 'cv', 
               number = folds)

rfFit <- train(Y ~ ., data = training, 
            method = "rf", 
            trControl = tc,
            maximize = TRUE,
            verbose = FALSE, ntree = 1000)

回复于 2024-05-07T08:11:23+08:00

插入包：列车功能的分层交叉验证

1 回答

相关问题