CommencerCommencer gratuitement

Créer des ensembles d’entraînement et de test

Avant de créer un modèle, il est essentiel de scinder votre jeu de données en deux : un ensemble d’entraînement qui servira à construire votre modèle de churn, et un ensemble de test qui servira à valider votre modèle. Pour cela, vous pouvez utiliser la fonction train_test_split() de sklearn.model_selection.

Vous allez vous exercer à créer des ensembles d’entraînement et de test. Le DataFrame telco est disponible dans votre espace de travail.

Cet exercice fait partie du cours

Marketing Analytics : prédire l’attrition client en Python

Afficher le cours

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# Import train_test_split
Modifier et exécuter le code