1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Machine Learning for Marketing in Python
Connected

cvičení

Rozbor pravidel rozhodovacího stromu

V tomto cvičení extrahuješ pravidla ve tvaru if-else z rozhodovacího stromu a vizualizuješ je, abys identifikoval/a hlavní faktory odchodu zákazníků.

Natrénovaná instance rozhodovacího stromu je načtená jako mytree a škálované příznaky jsou k dispozici jako pandas DataFrame s názvem train_X. Modul tree z knihovny sklearn a knihovna graphviz jsou již načteny.

Všimni si, že místo display(graph) používáme vlastní funkci display_image(), která ti usnadňuje zobrazení výstupu.

Pokyny

100 XP
  • Exportuj objekt graphviz z natrénovaného rozhodovacího stromu.
  • Přiřaď názvy příznaků.
  • Nastav přesnost na 1 a přidej názvy tříd.
  • Zavolej funkci Source() z knihovny graphviz a předej jí exportovaný objekt graphviz.