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
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Import train_test_split