Entraînez le régresseur GB
Vous allez maintenant entraîner le régresseur de gradient boosting gb
que vous avez instancié dans l'exercice précédent et prédire les étiquettes de l'ensemble de test.
L'ensemble de données est divisé en 80 % de formation et 20 % de test. Les matrices de caractéristiques X_train
et X_test
, ainsi que les tableaux y_train
et y_test
sont disponibles dans votre espace de travail. En outre, nous avons également chargé l'instance de modèle gb
que vous avez définie dans l'exercice précédent.
Cet exercice fait partie du cours
Apprentissage automatique avec des modèles arborescents en Python
Instructions
- Ajustez
gb
à l'ensemble de formation. - Prédire les étiquettes de l'ensemble de test et attribuer le résultat à
y_pred
.
Exercice interactif pratique
Essayez cet exercice en complétant cet exemple de code.
# Fit gb to the training set
____
# Predict test set labels
y_pred = ____