Začněte nyníZačněte zdarma

Natrénuj svůj první regresní strom

V tomto cvičení natrénuješ regresní strom pro předpovídání spotřeby mpg (miles per gallon) u automobilů z datasetu auto-mpg, a to s využitím všech šesti dostupných příznaků.

Dataset je předpřipravený a rozdělený na 80 % trénovacích a 20 % testovacích dat. Matice příznaků X_train a pole y_train jsou dostupné ve tvém pracovním prostředí.

Toto cvičení je součástí kurzu

Machine Learning with Tree-Based Models in Python

Zobrazit kurz

Pokyny k cvičení

  • Importuj DecisionTreeRegressor z sklearn.tree.
  • Vytvoř instanci DecisionTreeRegressor s názvem dt, maximální hloubkou 8 a hodnotou min_samples_leaf nastavenou na 0.13.
  • Natrénuj dt na trénovací sadě.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

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

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

# Fit dt to the training set
____.____(____, ____)
Upravit a spustit kód