開始使用免費開始

哪一個是主要預測因子?

你已經得到一個表現出色的預測結果,但主要的預測因子是哪些呢?要如何解讀模型,才能不只停留在原始輸出?機器學習模型常被批評不易解釋。不過,變數重要性排名能夠說明你所選特徵與目標之間的關聯度。現在就來檢視變數重要性,從這裡出發吧。

本練習屬於課程

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)
編輯並執行程式碼