Utwórz instancję modelu
W poniższej serii ćwiczeń zdiagnozujesz problemy obciążenia i wariancji drzewa regresji. Drzewo regresji zdefiniowane w tym ćwiczeniu posłuży do przewidywania spalania paliwa (mpg) samochodów ze zbioru danych auto, z wykorzystaniem wszystkich dostępnych cech.
Dane zostały już przetworzone — macierz cech X oraz tablica y są dostępne w twoim środowisku. Ponadto klasa DecisionTreeRegressor została zaimportowana z sklearn.tree.
To ćwiczenie jest częścią kursu
Uczenie maszynowe z modelami drzewiastymi w Pythonie
Instrukcje do ćwiczenia
- Zaimportuj
train_test_splitzsklearn.model_selection. - Podziel dane na 70% treningowych i 30% testowych.
- Utwórz instancję
DecisionTreeRegressorz maksymalną głębokością 4 i parametremmin_samples_leafustawionym na 0.26.
Interaktywne ćwiczenie praktyczne
Spróbuj tego ćwiczenia, uzupełniając ten przykładowy kod.
# 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)