CommencerCommencer gratuitement

Entraînez le classificateur AdaBoost

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

Une fois le classificateur ada formé, appelez la méthode .predict_proba() en passant X_test comme paramètre et extrayez 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 % de formation 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. En outre, nous avons également chargé le modèle instancié ada de l'exercice précédent.

Cet exercice fait partie du cours

Apprentissage automatique avec des modèles arborescents en Python

Afficher le cours

Instructions

  • Ajustez ada à l'ensemble de formation.

  • É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