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

View Course

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 = ____