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
Pokyny k cvičení
- Importuj
DecisionTreeRegressorzsklearn.tree. - Vytvoř instanci
DecisionTreeRegressors názvemdt, maximální hloubkou 8 a hodnotoumin_samples_leafnastavenou na 0.13. - Natrénuj
dtna 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
____.____(____, ____)