始める無料で始める

ROC/AUC スコアを計算する

Recall スコアは分類アルゴリズムの正確さを測るうえで重要ですが、False Negative の数を重視しすぎます。一方で、Precision は False Positive の数に重点を置きます。

これらを組み合わせた ROC 曲線を用いると、再現率と適合率の両方を評価できます。ROC 曲線の下の面積は AUC スコアとして計算されます。

この演習では、sklearn の roc_auc_score() 関数を使って初期モデルの ROC/AUC スコアを計算します。

features_testtarget_test の変数はワークスペース内で利用できます。

この演習はコースの一部です

HRアナリティクス:Pythonで従業員離職を予測する

コースを見る

演習の手順

  • ROC/AUC スコアを計算する関数をインポートします。
  • テストセットの特徴量に基づいて、初期モデルで離職を予測します。
  • 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)
コードを編集して実行