Export stromu

Při klasifikaci pomocí rozhodovacích stromů bývá přeučení (overfitting) nejčastěji důsledkem příliš hluboko rostlých stromů. Jak je vidět ze srovnání skóre přesnosti na trénovací a testovací sadě, tvé výsledky vykazují přeučení. To lze vyčíst i z vizualizace stromu.

V tomto cvičení exportuješ rozhodovací strom do textového souboru, který pak poslouží k vizualizaci.

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

HR Analytics: Predicting Employee Churn in Python

Zobrazit kurz

Pokyny k cvičení

Importuj funkci export_graphviz() ze submodulu sklearn.tree.
Natrénuj model na trénovacích datech.
Exportuj vizualizaci do souboru tree.dot.

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 the graphical visualization export function
from sklearn.____ import export_graphviz

# Apply Decision Tree model to fit Features to the Target
model.____(features_train,target_train)

# Export the tree to a dot file
____(model,"tree.____")

Upravit a spustit kód