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

cvičení

Vykresli ROC křivku

Vizualizace výkonu modelu pomocí ROC křivky ti umožňuje shrnout výkon modelu pro všechny možné prahy do jediného grafu. Zobrazuje citlivost (sensitivity) a specificitu (specificity) pro každý práh. Čím více je ROC křivka posunutá „nahoru a doleva", tím lepší model je.

Budeš předpovídat pravděpodobnosti příslušnosti ke třídám u zákazníků kreditních karet, kteří odešli, a výsledky zobrazíš jako ROC křivku.

Předpřipravený je model – rozhodovací strom natrénovaný na trénovací sadě zákazníků kreditních karet – a testovací data customers_test.

Pokyny

100 XP
  • Použij model k predikci pravděpodobností tříd na testovací sadě.
  • Výsledky přidej k testovací sadě pomocí bind_cols() a výsledek ulož jako predictions.
  • Vypočítej ROC křivku výsledku.
  • Vykresli ROC křivku pomocí autoplot().