Aan de slagGa gratis aan de slag

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

Cursus bekijken

Oefeninstructies

  • Importeer DecisionTreeRegressor uit sklearn.tree.
  • Instantieer een DecisionTreeRegressor dt met een maximale diepte van 8 en min_samples_leaf ingesteld op 0.13.
  • Fit dt op 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
____.____(____, ____)
Code bewerken en uitvoeren