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

exercițiu

Antrenează primul tău arbore de regresie

În acest exercițiu, vei antrena un arbore de regresie pentru a prezice consumul mpg (mile per galon) al mașinilor din setul de date auto-mpg, folosind toate cele șase caracteristici disponibile.

Setul de date a fost procesat în prealabil și împărțit în 80% antrenament și 20% testare. Matricea de caracteristici X_train și vectorul y_train sunt disponibile în spațiul tău de lucru.

Instrucțiuni

100 XP
  • Importă DecisionTreeRegressor din sklearn.tree.
  • Instanțiază un DecisionTreeRegressor dt cu adâncimea maximă 8 și min_samples_leaf setat la 0.13.
  • Antrenează dt pe setul de antrenament.