1. Učit se
    2. /
  2. Kurzy
    3. /
  3. Machine Learning s balíčkem caret v R
Connected

cvičení

Vytvoř vlastní trainControl

Dataset kvality vína byl příkladem regresní úlohy, teď se ale podíváme na klasifikaci. Jde o simulovaný dataset inspirovaný soutěží „don't overfit" na Kaggle z před několika lety.

Klasifikační úlohy jsou o něco složitější než regresní, protože funkci train() musíš předat vlastní summaryFunction, aby mohla k hodnocení modelů používat metriku AUC. Začni tím, že si vytvoříš vlastní trainControl, stejně jako v předchozí kapitole. Nezapomeň nastavit classProbs = TRUE – jinak twoClassSummary pro summaryFunction nebude fungovat správně.

Pokyny

100 XP

Pomocí funkce trainControl vytvoř vlastní objekt myControl pro klasifikaci.

  • Použij 10 foldů pro křížovou validaci.
  • Jako summaryFunction nastav twoClassSummary.
  • Nezapomeň nastavit classProbs = TRUE.