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
Instrukcje do ćwiczenia
- Zaimportuj
DecisionTreeRegressorzsklearn.tree. - Utwórz instancję
DecisionTreeRegressoro nazwiedtz maksymalną głębokością 8 i parametremmin_samples_leafustawionym na 0.13. - Dopasuj
dtdo 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
____.____(____, ____)