混同行列を作成する
ビデオで見たように、混同行列を使うとモデルの性能を評価できます。ただし、混同行列を作成する前に、予測確率をしきい値(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