What is a stratified split and when do you need it?
PICTURE THIS: DATA SPLIT
Fit on train, tune on val, report on test once.
Simple meaning
A stratified split keeps class proportions similar in train, val, and test.
WHY — Train/Val/Test instead of guessing?
Why interviewers care about Train/Val/Test:
who only read docs from people who shipped.
and tied to AI / ML work.
Name the idea, why it exists, then one short example.
End with when you use it and one common pitfall.
STEPS — What happens step by step?
Before you speak the answer, walk the interviewer through these steps:
- 1A stratified split keeps
class proportions similar in train, val, and test.
- 2You need it for
classification, especially with rare classes, so a fold is not accidentally all majority.
- 3For regression you can
stratify on binned targets when the y distribution is skewed.
- 4Give an example
One tiny concrete case you can say aloud.
- 5Common mistake
What juniors usually get wrong.
- 6Close
When you pick this over the alternative.
EXAMPLE — See it in action
Here's a short line you can speak, broken into clear beats:
Note: Adapt this scaffold to your own project — keep it under 60–90 seconds.
Key takeaway
A stratified split keeps class proportions similar in train, val, and test. You need it for classification, especially with rare classes, so a fold is not accidentally all majority.