AI engineering glossary · Feature engineering data prep

What is group-aware splitting?

Group-aware splitting (also called GroupKFold): Splitting data so that all rows belonging to one group, such as a patient or customer, fall entirely in either train or test. It prevents leakage caused by the same entity appearing on both sides.

Level: advancedTopic: Feature engineering data prep

Group-aware splitting: an example

With several scans per patient, all scans of one patient go to the same fold, so the model is tested on unseen patients.

Related terms

More Feature engineering data prep terms

Where you learn it

Group-aware splitting is part of the AI Engineering Program, which teaches it with notes, a hands-on lab and an exam. The first lesson is free.

Before you go

Take the cheat sheet with you

One free PDF: the key AI-engineering terms, formulas and Python idioms. No spam; unsubscribe any time.