1. Learn
    2. /
  2. Kurser
    3. /
  3. Maskininlärning med trädbaserade modeller i Python
Connected

övning

Instantiera modellen

I följande övningar ska du diagnostisera bias- och variansproblem i ett regressionsträd. Det regressionsträd du definierar i den här övningen används för att förutsäga bränsleförbrukningen (mpg) för bilar i auto-datamängden, med hjälp av alla tillgängliga särdrag.

Vi har redan bearbetat data och laddat feature-matrisen X och arrayen y i din workspace. Dessutom har klassen DecisionTreeRegressor importerats från sklearn.tree.

Instruktioner

100 XP
  • Importera train_test_split från sklearn.model_selection.
  • Dela upp data i 70% träning och 30% test.
  • Instantiera en DecisionTreeRegressor med ett maximalt djup på 4 och min_samples_leaf satt till 0.26.