1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Machine Learning with Tree-Based Models in Python
Connected

cvičení

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í.

Pokyny

100 XP
  • 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ě.