Знову про glmnet

Machine Learning з пакетом caret в R

Zach Mayer

Data Scientist at DataRobot and co-author of caret

Огляд glmnet

  • Лінійна модель із вбудованим відбором змінних
  • Потужна базова модель
  • Переваги
    • Швидко навчається
    • Ігнорує шумові змінні
    • Дає інтерпретовані коефіцієнти
Machine Learning з пакетом caret в R

Приклад: glmnet на даних відтоку

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 з пакетом caret в R

Візуалізація результатів

plot(model_glmnet)

Зображення: лінійний графік ROC проти параметра регуляризації для двох значень частки змішування 0 та 1. ROC значно вищий, коли частка змішування дорівнює 1.

Machine Learning з пакетом caret в R

Графік коефіцієнтів

plot(model_glmnet$finalModel)

Зображення: спеціальний графік glmnet — коефіцієнти проти L1-норми

Machine Learning з пакетом caret в R

Давайте потренуємось!

Machine Learning з пакетом caret в R

Preparing Video For Download...