Train the SGB regressor

In this exercise, you'll train the SGBR sgbr instantiated in the previous exercise and predict the test set labels.

The bike sharing demand dataset is already loaded processed for you; it is split into 80% train and 20% test. The feature matrices X_train and X_test, the arrays of labels y_train and y_test, and the model instance sgbr that you defined in the previous exercise are available in your workspace.

This exercise is part of the course

Machine Learning with Tree-Based Models in Python

View Course

Exercise instructions

  • Fit sgbr to the training set.
  • Predict the test set labels and assign the results to y_pred.

Hands-on interactive exercise

Have a go at this exercise by completing this sample code.

# Fit sgbr to the training set
____

# Predict test set labels
y_pred = ____