開始使用免費開始

建立一個混淆矩陣

正如你在影片中看到的,混淆矩陣可以用來判斷模型的表現如何。不過,在建立混淆矩陣之前,你需要先用一個「臨界值」把預測機率分類為 10

注意:1 代表 Inactive,0 代表 Active。

工作區中已提供 prediction_test,其中包含 test_set 中所有案例的離職(turnover)預測機率。

本練習屬於課程

HR 分析:以 R 預測員工流失

檢視課程

練習說明

  • 以 0.5 為臨界值,將 prediction_test 中的數值型預測轉換為類別型預測向量。
  • 使用 prediction_categories 與測試集的實際值(test_set$turnover)建立混淆矩陣。

動手互動練習

試著完成這個範例程式碼,體驗一下這個練習。

# Classify predictions using a cut-off of 0.5
prediction_categories <- ___(prediction_test > 0.5, 1, 0)

# Construct a confusion matrix
conf_matrix <- ___(prediction_categories, test_set$turnover)
conf_matrix
編輯並執行程式碼