Zacznij terazZacznij za darmo

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.

To ćwiczenie jest częścią kursu

Uczenie maszynowe z modelami drzewiastymi w Pythonie

Zobacz kurs

Instrukcje do ćwiczenia

  • 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.

Interaktywne ćwiczenie praktyczne

Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.

# Import DecisionTreeRegressor from sklearn.tree
from ____.____ import ____

# Instantiate dt
____ = ____(max_depth=____,
             ____=____,
            random_state=3)

# Fit dt to the training set
____.____(____, ____)
Edytuj i uruchom kod