CommencerCommencer gratuitement

Entraîner le classificateur AdaBoost

Maintenant que vous avez instancié le classificateur AdaBoost ada, il est temps de l'entraîner. Vous devrez également prédire les probabilités d'obtenir la classe positive dans l'ensemble de test. Cela peut être réalisé de la manière suivante :

Une fois que le classificateur ada a été formé, veuillez appeler la méthode .predict_proba() en passant X_test comme paramètre et extraire ces probabilités en découpant toutes les valeurs de la deuxième colonne comme suit :

ada.predict_proba(X_test)[:,1]

L'ensemble de données Indian Liver est traité pour vous et divisé en 80 % d'entraînement et 20 % de 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 également chargé le ada, modèle instancié de l'exercice précédent.

Cet exercice fait partie du cours

Machine learning avec des modèles arborescents en Python

Afficher le cours

Instructions

  • Ajustez l'ada e à l'ensemble d'apprentissage.

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

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# Fit ada to the training set
____

# Compute the probabilities of obtaining the positive class
y_pred_proba = ____.____(____)[____]
Modifier et exécuter le code