ÎncepețiÎncepe gratuit

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.

Acest exercițiu face parte din cursul

Machine Learning cu modele bazate pe arbori în Python

Vezi cursul

Instrucțiuni pentru exercițiu

  • 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.

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

# Import DecisionTreeRegressor from sklearn.tree
from ____.____ import ____

# Instantiate dt
____ = ____(max_depth=____,
             ____=____,
            random_state=3)

# Fit dt to the training set
____.____(____, ____)
Editează și rulează codul