Caret Paketi: Tren Fonksiyonunda Tabakalı Çapraz Doğrulama

Question

Büyük bir dengesiz veri setine bir modele uyması için tren fonksiyonunu kullanırken katmanlaştırılmış çapraz doğrulamayı gerçekleştirmenin bir yolu var mı? İleri doğru biliyorum, katlama çapraz doğrulaması mümkündür ama kategorilerim oldukça dengesiz. Bu konu hakkında tartışma gördüm ama gerçek bir kesin cevap yok.

Şimdiden teşekkürler.

Answer 1

Kullanıcıya dizinin çapraz doğrulamayı yapmasını belirleyebilen 'dizin' adlı bir parametre var.

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)