Kom igångKom igång gratis

Träna ditt första regressionsträd

I den här övningen tränar du ett regressionsträd för att förutsäga mpg-förbrukningen (miles per gallon) för bilar i auto-mpg-datamängden med hjälp av alla sex tillgängliga särdrag.

Datamängden är förberedd åt dig och delad i 80% träning och 20% test. Feature-matrisen X_train och arrayen y_train finns tillgängliga i din arbetsyta.

Den här övningen är en del av kursen

Maskininlärning med trädbaserade modeller i Python

Visa kurs

Övningsinstruktioner

  • Importera DecisionTreeRegressor från sklearn.tree.
  • Skapa en instans av DecisionTreeRegressor med namnet dt, med maxdjup 8 och min_samples_leaf satt till 0.13.
  • Anpassa dt till träningsdata.

Interaktiv övning med praktiskt arbete

Testa den här övningen genom att slutföra den här exempelkoden.

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

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

# Fit dt to the training set
____.____(____, ____)
Redigera och kör kod