A team trains a GPU-accelerated XGBoost classifier on an imbalanced churn dataset and wants a dependable estimate of how it will generalise. They have a single GPU with enough memory to fit the data several times over, and a colleague proposes plain random 5-fold cross-validation. Given the imbalance, which adjustment to the resampling scheme most directly improves the reliability of the per-fold metrics?
- AReplace the cross-validation with one larger 90/10 holdout split, since a single bigger test set is more representative than averaging several small folds.
- BRaise the number of folds from 5 to 20 so each training fold sees more data, which on its own balances the class ratio across folds.
- CUse stratified k-fold cross-validation so each fold preserves roughly the same minority-class proportion as the full dataset. Correct
- DShuffle the rows before splitting into the 5 folds, because randomising the order alone balances the class ratio within every fold.
Why A is wrong: Tempting because a larger test set feels more stable, but a single split discards the variance reduction of averaging across folds and may leave the rare class barely represented, so it does not address the imbalance.
Why B is wrong: Tempting because larger k does enlarge each training portion, but it shrinks each test fold and can leave individual folds with no minority cases, so increasing k alone does not guarantee balanced representation.
Why C is correct: Correct: stratification holds the rare-class ratio constant across folds, preventing folds with too few or zero positive cases, which stabilises the per-fold metrics and yields a trustworthy averaged estimate under imbalance.
Why D is wrong: Tempting because shuffling removes ordering artefacts, but random shuffling only balances folds in expectation and can still produce folds where the rare class is severely under-represented, which stratification explicitly prevents.