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
<cours>Marketing Analytics : prédire l’attrition client en Python</cours>Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Import train_test_split