CommencezCommencez gratuitement

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.

Cette activité fait partie du cours

Apprentissage supervisé avec scikit-learn

Voir le cours

Instructions de l’exercice

  • 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.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# Import the module
from ____ import ____

X = churn_df.drop("churn", axis=1).values
y = churn_df["churn"].values

# Split into training and test sets
X_train, X_test, y_train, y_test = ____(____, ____, test_size=____, random_state=____, stratify=____)
knn = KNeighborsClassifier(n_neighbors=5)

# Fit the classifier to the training data
____

# Print the accuracy
print(knn.score(____, ____))
Modifier et exécuter le code