1. Apprendre
    2. /
  2. Cours
    3. /
  3. Modèles linéaires généralisés en Python
Connected

Exercice

Calculer la matrice de confusion

Comme vous l'avez vu dans la vidéo, le modèle de régression logistique produit deux types de prédictions : une prédiction continue, sous forme de probabilité, et une prédiction de classe qui, dans l'exemple de l'ensemble de données wells, correspond à une catégorie discrète avec deux classes.

Dans l'exercice précédent, vous avez calculé la prédiction continue sous forme de probabilité. Dans cet exercice, vous utiliserez ces valeurs pour attribuer une classe à chaque observation de votre échantillon wells_test. Enfin, vous décrirez le modèle en utilisant la matrice de confusion.

Les prédictions calculées prediction et wells_test sont chargées dans votre espace de travail.

Instructions 1/3

undefined XP
    1
    2
    3
  • À partir des prédictions calculées prediction, classez-les en étiquettes de classes 0 et 1 en utilisant une valeur seuil de 0.5, puis enregistrez le résultat dans y_prediction.