1. Learn
    2. /
  2. Courses
    3. /
  3. Machine Learning with Tree-Based Models in Python

Exercise

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.

Instructions

100 XP
  • Fit gb to the training set.
  • Predict the test set labels and assign the result to y_pred.