Entraîner le régresseur GB
Vous allez maintenant entraîner le régresseur à boost gradient 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 % d'entraînement 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. De plus, 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
Machine learning avec des modèles arborescents en Python
Instructions
- Ajustez l'
gbe à l'ensemble d'apprentissage. - Veuillez 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 = ____