НачатьНачать бесплатно

Какой признак является главным предиктором?

Вы получили отличный прогноз — но какие признаки сыграли ключевую роль? Как разобраться в том, что происходит внутри модели, и выйти за рамки сырых результатов? Модели машинного обучения нередко критикуют за недостаточную интерпретируемость. Тем не менее ранжирование по важности переменных помогает понять, насколько выбранные признаки связаны с целевой переменной. Давайте исследуем важность переменных и разберёмся, что за этим стоит.

Это упражнение является частью курса

Конструирование признаков в R

Посмотреть курс

Инструкции к упражнению

  • Постройте график важности переменных.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

lr_fit <- lr_workflow %>%
  fit(test)

lr_aug <- lr_fit %>%
  augment(test)

lr_aug %>% class_evaluate(truth = Attrition,
                          estimate = .pred_class,
                          .pred_No)

# Create a variable importance chart
lr_fit %>%
  extract_fit_parsnip() %>%
  ___(aesthetics = list(fill = "steelblue"), num_features = 5)
Редактировать и запускать код