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
Instrucțiuni pentru exercițiu
- Importă
DecisionTreeRegressordinsklearn.tree. - Instanțiază un
DecisionTreeRegressordtcu adâncimea maximă 8 șimin_samples_leafsetat la 0.13. - Antrenează
dtpe 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
____.____(____, ____)