CommencezCommencez gratuitement

Évaluer un classificateur de prédiction du diabète

Dans ce chapitre, vous allez travailler avec l'ensemble de données diabetes_df présenté plus tôt.

L'objectif est de prédire si une personne est susceptible ou non d'avoir le diabète à partir des caractéristiques indice de masse corporelle (IMC) et âge (en années). Il s'agit donc d'un problème de classification binaire. Une valeur cible de 0 indique que la personne n'a PAS le diabète, tandis qu'une valeur de 1 indique que la personne A le diabète.

diabetes_df a été préchargé pour vous comme DataFrame pandas et divisé en X_train, X_test, y_train et y_test. De plus, un KNeighborsClassifier() a été instancié et affecté à knn.

Vous allez ajuster le modèle, faire des prédictions sur l'ensemble de test, puis produire une matrice de confusion et un rapport de classification.

Cette activité fait partie du cours

Apprentissage supervisé avec scikit-learn

Voir le cours

Instructions de l’exercice

  • Importez confusion_matrix et classification_report.
  • Ajustez le modèle aux données d'entraînement.
  • Prédisez les étiquettes de l'ensemble de test et enregistrez le résultat dans y_pred.
  • Calculez et affichez la matrice de confusion et le rapport de classification pour les étiquettes de test par rapport aux étiquettes prédites.

Exercice interactif pratique

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

# Import confusion matrix
____

knn = KNeighborsClassifier(n_neighbors=6)

# Fit the model to the training data
____

# Predict the labels of the test data: y_pred
y_pred = ____

# Generate the confusion matrix and classification report
print(____(____, ____))
print(____(____, ____))
Modifier et exécuter le code