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

Создание экземпляра модели

В следующей серии упражнений вы будете диагностировать проблемы смещения и дисперсии у дерева регрессии. Дерево регрессии, которое вы определите в этом упражнении, будет использоваться для предсказания расхода топлива (mpg) автомобилей из набора данных auto на основе всех доступных признаков.

Данные уже обработаны: матрица признаков X и массив y загружены в вашем рабочем пространстве. Кроме того, класс DecisionTreeRegressor импортирован из sklearn.tree.

Это упражнение является частью курса

Машинное обучение на основе древовидных моделей в Python

Посмотреть курс

Инструкции к упражнению

  • Импортируйте train_test_split из sklearn.model_selection.
  • Разделите данные в соотношении 70% на обучение и 30% на тест.
  • Создайте экземпляр DecisionTreeRegressor с максимальной глубиной 4 и параметром min_samples_leaf, равным 0.26.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

# Import train_test_split from sklearn.model_selection
____

# Set SEED for reproducibility
SEED = 1

# Split the data into 70% train and 30% test
X_train, X_test, y_train, y_test = ____(____, ____, test_size=____, random_state=SEED)

# Instantiate a DecisionTreeRegressor dt
dt = ____(____=____, ____=____, random_state=SEED)
Редактировать и запускать код