Train je eerste regressieboom
In deze oefening train je een regressieboom om het mpg-verbruik (miles per gallon) van auto's in de auto-mpg dataset te voorspellen met alle zes beschikbare features.
De gegevensset is alvast voor je verwerkt en opgesplitst in 80% train en 20% test. De feature-matrix X_train en de array y_train staan klaar in je werkruimte.
Deze oefening maakt deel uit van de cursus
Machine Learning met boomgebaseerde modellen in Python
Oefeninstructies
- Importeer
DecisionTreeRegressoruitsklearn.tree. - Instantieer een
DecisionTreeRegressordtmet een maximale diepte van 8 enmin_samples_leafingesteld op 0.13. - Fit
dtop de trainingsset.
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
# Import DecisionTreeRegressor from sklearn.tree
from ____.____ import ____
# Instantiate dt
____ = ____(max_depth=____,
____=____,
random_state=3)
# Fit dt to the training set
____.____(____, ____)