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