Gói Caret: Xác nhận chữ thập phân cấp trong chức năng đào tạo

Question

Có cách nào để thực hiện xác nhận chéo phân tầng khi sử dụng chức năng đào tạo để phù hợp với mô hình cho một tập dữ liệu mất cân bằng lớn không? Tôi biết thẳng về phía trước k fold xác nhận chéo là có thể nhưng thể loại của tôi là rất không cân bằng. Tôi đã nhìn thấy cuộc thảo luận về chủ đề này nhưng không có câu trả lời dứt khoát thực sự.

Xin cảm ơn trước.

Answer 1

Có một tham số được gọi là 'chỉ mục' có thể cho phép người dùng chỉ định chỉ mục thực hiện xác thực chéo.

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)