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
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 = ____.____(____)[____]