1. Learn
    2. /
  2. Курси
    3. /
  3. Машинне навчання з деревоподібними моделями в Python
Connected

вправа

Навчіть свій перший регресійний дерево

У цій вправі ви навчите регресійне дерево прогнозувати mpg (миль на галон) для автомобілів у наборі даних auto-mpg, використовуючи всі шість наявних ознак.

Набір даних уже підготовлено та розділено: 80% — тренування, 20% — тест. Матриця ознак X_train і масив y_train доступні у вашому робочому середовищі.

Інструкції

100 XP
  • Імпортуйте DecisionTreeRegressor з sklearn.tree.
  • Створіть DecisionTreeRegressor dt з максимальною глибиною 8 і min_samples_leaf, встановленим на 0.13.
  • Навчіть dt на тренувальній вибірці.