1. Apprendre
    2. /
  2. Cours
    3. /
  3. Apprentissage supervisé avec scikit-learn
Connected

Exercice

Découpage entraînement/test + calcul de la justesse

C'est le moment de vous exercer à diviser vos données en ensembles d'entraînement et de test avec le jeu de données churn_df!

Des tableaux NumPy ont été créés pour vous : les caractéristiques dans X et la variable cible dans y.

Instructions

100 XP
  • Importez train_test_split depuis sklearn.model_selection.
  • Divisez X et y en ensembles d'entraînement et de test, avec test_size à 20 %, random_state à 42, et en veillant à ce que la proportion des étiquettes cibles reflète celle du jeu de données d'origine.
  • Ajustez le modèle knn sur les données d'entraînement.
  • Calculez et affichez la justesse du modèle sur les données de test.