开始使用免费开始使用

主要预测因子是哪一个?

您的预测效果非常出色,但主要的预测因子是什么呢?如何理解模型,从而不仅停留在原始结果之上?机器学习模型常被批评可解释性不足。不过,变量重要性排序能够揭示您所选特征与结果之间的相关性。让我们来分析变量重要性,并以此为起点。

本练习是课程的一部分

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)
编辑并运行代码