開始使用免費開始

計算 ROC/AUC 分數

雖然 Recall 分數是衡量分類演算法準確度的重要指標,但它對「假陰性(False Negatives)」的數量權重較高;相對地,Precision 則較著重「假陽性(False Positives)」的數量。

將兩者結合得到的 ROC 曲線,可以同時衡量 recall 與 precision。而 ROC 曲線下的面積(AUC)則作為整體表現的分數。

在這個練習中,你將使用 sklearn 的 roc_auc_score() 函式,計算初始模型的 ROC/AUC 分數。

變數 features_testtarget_test 已在你的工作環境中可用。

本練習屬於課程

HR 分析:用 Python 預測員工流失

檢視課程

練習說明

  • 匯入用來計算 ROC/AUC 分數的函式。
  • 使用初始模型,根據測試集的特徵來預測離職(churn)。
  • 比較 target_test 與預測結果,計算 ROC/AUC 分數。

動手互動練習

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

# Import the function to calculate ROC/AUC score
from sklearn.____ import ____

# Use initial model to predict churn (based on features_test)
prediction = model.predict(____)

# Calculate ROC/AUC score by comparing target_test with the prediction
____(____, prediction)
編輯並執行程式碼