CommencezCommencez gratuitement

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

Voir le cours

Instructions de l’exercice

  • Ajustez gb sur 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 = ____
Modifier et exécuter le code