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

Connected

ćwiczenie

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.

Instrukcje

100 XP
  • Zaimportuj train_test_split z sklearn.model_selection.
  • Podziel dane na 70% treningowych i 30% testowych.
  • Utwórz instancję DecisionTreeRegressor z maksymalną głębokością 4 i parametrem min_samples_leaf ustawionym na 0.26.