1. Apprendre
    2. /
  2. Cours
    3. /
  3. Machine Learning avec des modèles à base d'arbres en Python
Connected

Exercice

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.

Instructions

100 XP
  • Ajustez gb sur l'ensemble d'entraînement.
  • Prédisez les étiquettes de l'ensemble de test et assignez le résultat à y_pred.