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
Instructions de l’exercice
- Importez
train_test_splitdepuissklearn.model_selection. - Divisez
Xetyen ensembles d'entraînement et de test, avectest_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
knnsur 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(____, ____))