Reintroducere glmnet

Machine Learning cu caret în R

Zach Mayer

Data Scientist at DataRobot and co-author of caret

Recapitulare glmnet

  • Model liniar cu selecție automată a variabilelor
  • Model de referință excelent
  • Avantaje
    • Antrenare rapidă
    • Ignoră variabilele zgomotoase
    • Coeficienți ușor de interpretat
Machine Learning cu caret în R

Exemplu: glmnet pe datele 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 cu caret în R

Vizualizarea rezultatelor

plot(model_glmnet)

Imaginea prezintă un grafic liniar al curbei ROC în funcție de parametrul de regularizare, pentru două valori ale procentului de mixare: 0 și 1. ROC este mult mai mare când procentul de mixare este 1.

Machine Learning cu caret în R

Reprezentarea coeficienților

plot(model_glmnet$finalModel)

Imaginea prezintă un grafic glmnet personalizat al coeficienților în funcție de norma L1

Machine Learning cu caret în R

Să exersăm!

Machine Learning cu caret în R

Preparing Video For Download...