Создание экземпляра модели
В следующей серии упражнений вы будете диагностировать проблемы смещения и дисперсии у дерева регрессии. Дерево регрессии, которое вы определите в этом упражнении, будет использоваться для предсказания расхода топлива (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)