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

cvičení

Natrénuj rozhodovací strom

Teď si vyzkoušíš sestavit model rozhodovacího stromu. Rozhodovací strom je sada strojově naučených pravidel typu „if-else", která v případě churn analýzy telekomunikační společnosti rozhodují o tom, zda zákazník odejde, nebo ne. Níže vidíš příklad rozhodovacího stromu sestaveného na základě slavné datové sady Titanic o přežití cestujících.

Proměnné train_X, test_X, train_Y, test_Y z předchozího cvičení jsou už načtené. Stejně tak modul tree a funkce accuracy_score z knihovny sklearn. Teď sestavíš vlastní model a otestuješ, jak si vede na datech, která dosud neviděl.

Pokyny

100 XP
  • Inicializuj model rozhodovacího stromu s hodnotou max_depth nastavenou na 5.
  • Natrénuj model na trénovacích datech — nejprve předej train_X, pak train_Y.
  • Proveď predikci na testovacích datech, tedy na test_X.
  • Vyhodnoť výkon modelu na testovacích datech porovnáním skutečných testovacích štítků s predikovanými.