Обучение регрессора на основе градиентного бустинга
Теперь вы обучите регрессор градиентного бустинга gb, который создали в предыдущем упражнении, и получите предсказания для тестовой выборки.
Набор данных разбит на обучающую (80%) и тестовую (20%) части. Матрицы признаков X_train и X_test, а также массивы y_train и y_test уже доступны в вашей рабочей среде. Кроме того, там же загружен экземпляр модели gb, определённый в предыдущем упражнении.
Это упражнение является частью курса
Машинное обучение на основе древовидных моделей в Python
Инструкции к упражнению
- Обучите
gbна обучающей выборке. - Получите предсказания для тестовой выборки и сохраните результат в переменную
y_pred.
Интерактивное практическое упражнение
Попробуйте выполнить это упражнение, дополнив этот пример кода.
# Fit gb to the training set
____
# Predict test set labels
y_pred = ____