Train the GB regressor
You'll now train the gradient boosting regressor gb
that you instantiated in the previous exercise and predict test set labels.
The dataset is split into 80% train and 20% test. Feature matrices X_train
and X_test
, as well as the arrays y_train
and y_test
are available in your workspace. In addition, we have also loaded the model instance gb
that you defined in the previous exercise.
This exercise is part of the course
Machine Learning with Tree-Based Models in Python
Exercise instructions
- Fit
gb
to the training set. - Predict the test set labels and assign the result to
y_pred
.
Hands-on interactive exercise
Have a go at this exercise by completing this sample code.
# Fit gb to the training set
____
# Predict test set labels
y_pred = ____