1. Nauka
    2. /
  2. Kursy
    3. /
  3. Uczenie maszynowe z modelami drzewiastymi w Pythonie
Connected

ćwiczenie

Trenowanie pierwszego drzewa regresji

W tym ćwiczeniu wytrenuj drzewo regresji, które będzie przewidywać wartość mpg (mile na galon) dla samochodów ze zbioru danych auto-mpg, korzystając ze wszystkich sześciu dostępnych cech.

Zbiór danych został już przetworzony i podzielony na 80% zbiór treningowy i 20% zbiór testowy. Macierz cech X_train oraz tablica y_train są dostępne w twoim środowisku pracy.

Instrukcje

100 XP
  • Zaimportuj DecisionTreeRegressor z sklearn.tree.
  • Utwórz instancję DecisionTreeRegressor o nazwie dt z maksymalną głębokością 8 i parametrem min_samples_leaf ustawionym na 0.13.
  • Dopasuj dt do zbioru treningowego.