始める無料で始める

混同行列を作成する

ビデオで見たように、混同行列を使うとモデルの性能を評価できます。ただし、混同行列を作成する前に、予測確率をしきい値(cut-off)1 または 0 に分類する必要があります。

注: 1 は Inactive、0 は Active を意味します。

test_set のすべてのケースに対する離職予測確率を含む prediction_test がワークスペースに用意されています。

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

HRアナリティクス:Rで社員離職を予測する

コースを見る

演習の手順

  • prediction_test の数値予測を、しきい値 0.5 を用いてカテゴリ予測のベクトルに変換します。
  • 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
コードを編集して実行