1. Learn
    2. /
  2. Cursuri
    3. /
  3. Machine Learning cu modele bazate pe arbori în Python
Connected

exercițiu

Instanțiază modelul

În această serie de exerciții, vei diagnostica problemele de bias și varianță ale unui arbore de regresie. Arborele de regresie pe care îl vei defini în acest exercițiu va fi folosit pentru a prezice consumul de carburant (mpg) al mașinilor din setul de date auto, utilizând toate caracteristicile disponibile.

Am procesat deja datele și am încărcat matricea de caracteristici X și vectorul y în spațiul tău de lucru. În plus, clasa DecisionTreeRegressor a fost importată din sklearn.tree.

Instrucțiuni

100 XP
  • Importă train_test_split din sklearn.model_selection.
  • Împarte datele în 70% antrenament și 30% testare.
  • Instanțiază un DecisionTreeRegressor cu adâncimea maximă 4 și min_samples_leaf setat la 0.26.