1. Apprendre
    2. /
  2. Cours
    3. /
  3. Machine Learning avec des modèles à base d'arbres en Python
Connected

Exercice

Entraîner le classificateur AdaBoost

Maintenant que vous avez instancié le classificateur AdaBoost ada, il est temps de l'entraîner. Vous allez aussi prédire les probabilités d'obtenir la classe positive dans l'ensemble de test. Voici comment procéder :

Une fois le classificateur ada entraîné, appelez la méthode .predict_proba() en passant X_test en paramètre et extrayez ces probabilités en sélectionnant toutes les valeurs de la deuxième colonne comme suit :

ada.predict_proba(X_test)[:,1]

Le jeu de données Indian Liver a été préparé pour vous et divisé en 80 % pour l'entraînement et 20 % pour le test. Les matrices de caractéristiques X_train et X_test, ainsi que les tableaux d'étiquettes y_train et y_test, sont disponibles dans votre espace de travail. De plus, nous avons déjà chargé le modèle instancié ada de l'exercice précédent.

Instructions

100 XP

  • Ajustez ada sur l'ensemble d'entraînement.

  • Évaluez les probabilités d'obtenir la classe positive dans l'ensemble de test.