Entraîner le régressseur GB
Vous allez maintenant entraîner le régressseur à gradient boosting gb que vous avez instancié à l'exercice précédent et prédire les étiquettes de l'ensemble de test.
Le jeu de données est séparé en 80 % pour l'entraînement et 20 % pour le 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. De plus, nous avons aussi chargé l'instance du modèle gb que vous avez définie à l'exercice précédent.
Cette activité fait partie du cours
Machine Learning avec des modèles à base d'arbres en Python
Instructions de l’exercice
- Ajustez
gbsur l'ensemble d'entraînement. - Prédisez les étiquettes de l'ensemble de test et assignez le résultat à
y_pred.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Fit gb to the training set
____
# Predict test set labels
y_pred = ____