Le découpage train/test sépare les données en un jeu d'entraînement, pour ajuster le modèle, et un jeu de test, gardé de côté pour estimer sa performance sur des données nouvelles. On fixe random_state pour la reproductibilité et on utilise stratify en classification pour conserver les proportions des classes.
Exemple
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)