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

cvičení

Vytvoření instance modelu

V následující sérii cvičení budeš diagnostikovat problémy s biasem a variancí regresního stromu. Regresní strom, který definuješ v tomto cvičení, použiješ k předpovídání spotřeby mpg automobilů z datasetu auto na základě všech dostupných příznaků.

Data jsme už zpracovali a do tvého pracovního prostředí načetli matici příznaků X a pole y. Navíc byla třída DecisionTreeRegressor importována z sklearn.tree.

Pokyny

100 XP
  • Importuj train_test_split z sklearn.model_selection.
  • Rozděl data na 70 % trénovací a 30 % testovací část.
  • Vytvoř instanci DecisionTreeRegressor s maximální hloubkou 4 a hodnotou min_samples_leaf nastavenou na 0.26.