Entraîner le régressseur SGB
Dans cet exercice, vous allez entraîner le SGBR sgbr instancié dans l'exercice précédent et prédire les étiquettes de l'ensemble de test.
Le jeu de données sur la demande de vélos en libre-service est déjà chargé et prétraité pour vous; il est divisé en 80 % pour l'entraînement et 20 % pour le test. Les matrices de variables X_train et X_test, les tableaux d'étiquettes y_train et y_test, ainsi que l'instance du modèle sgbr que vous avez définie à l'exercice précédent sont disponibles dans votre espace de travail.
Cette activité fait partie du cours
Machine Learning avec des modèles à base d'arbres en Python
Instructions de l’exercice
- Ajustez
sgbrsur 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 sgbr to the training set
____
# Predict test set labels
y_pred = ____