НачатьНачать бесплатно

Обучение регрессора на основе градиентного бустинга

Теперь вы обучите регрессор градиентного бустинга 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 = ____
Редактировать и запускать код