CommencerCommencer gratuitement

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

Afficher le cours

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