Arbres pour prédire les défauts
Vous allez maintenant entraîner un modèle d'arbres à renforcement de gradient sur les données de crédit et consulter un échantillon de prédictions. Vous souvenez-vous de votre première observation des prédictions du modèle de régression logistique? Elles n'étaient pas très convaincantes. Pensez-vous que ce modèle donnera de meilleurs résultats?
Les données de crédit cr_loan_prep, les ensembles d'entraînement X_train et y_train, ainsi que les données de test X_test sont déjà disponibles dans l'espace de travail. Le paquet XGBoost est chargé sous le nom xgb.
Cette activité fait partie du cours
Modélisation du risque de crédit en Python
Instructions de l’exercice
- Créez et entraînez un arbre à renforcement de gradient avec
XGBClassifier()et nommez-leclf_gbt. - Prévoyez les probabilités de défaut sur les données de test et stockez les résultats dans
gbt_preds. - Créez deux trames de données,
preds_dfettrue_df, pour conserver les cinq premières prédictions et les vraies valeursloan_status. - Concaténez puis affichez les trames de données
true_dfetpreds_df, dans cet ordre, et vérifiez les résultats du modèle.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Train a model
import xgboost as xgb
____ = xgb.____().fit(____, np.ravel(____))
# Predict with a model
____ = clf_gbt.____(____)
# Create dataframes of first five predictions, and first five true labels
____ = pd.DataFrame(____[:,1][0:5], columns = ['prob_default'])
____ = y_test.____()
# Concatenate and print the two data frames for comparison
print(pd.____([____.reset_index(drop = True), ____], axis = 1))