Opět glmnet

Machine Learning s balíčkem caret v R

Zach Mayer

Data Scientist at DataRobot and co-author of caret

Přehled glmnet

  • Lineární model s vestavěným výběrem proměnných
  • Skvělý základní model
  • Výhody
    • Rychlé trénování
    • Ignoruje šumové proměnné
    • Poskytuje interpretovatelné koeficienty
Machine Learning s balíčkem caret v R

Příklad: glmnet na datech churn

set.seed(42)
model_glmnet <- train(
  churn ~ ., 
  churnTrain,
  metric = "ROC",
  method = "glmnet",
  tuneGrid = expand.grid(
    alpha = 0:1, 
    lambda = 0:10 / 10
  ),
  trControl = myControl
)
Machine Learning s balíčkem caret v R

Vizualizace výsledků

plot(model_glmnet)

Obrázek zobrazuje spojnicový graf ROC křivky v závislosti na regularizačním parametru pro dvě hodnoty podílu míchání 0 a 1. ROC je výrazně vyšší při podílu míchání 1.

Machine Learning s balíčkem caret v R

Graf koeficientů

plot(model_glmnet$finalModel)

Obrázek zobrazuje vlastní graf glmnet koeficientů v závislosti na normě L1

Machine Learning s balíčkem caret v R

Pojďme si procvičit!

Machine Learning s balíčkem caret v R

Preparing Video For Download...