始める無料で始める

単一モデルの予測

分類モデルの性能を計算するには、Attrition の実際の値とモデルが予測した値を比較する必要があります。 二値分類(たとえば precision や recall の計算)の指標を算出する際は、実測値と予測値のベクトルを二値(TRUE/FALSE)に変換します。

この演習では、最初の交差検証フォールドの model と validate データフレームを例に、これらのベクトルをどのように準備するかを学びます。

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

Tidyverse で学ぶ Machine Learning

コースを見る

演習の手順

  • 交差検証の最初のフォールドから modelvalidate データフレームを取り出します。
  • validate データフレームの Attrition 列を抽出し、値を二値(TRUE/FALSE)に変換します。
  • model を使って、validate データフレームに対する離職確率を予測します。
  • 予測確率を二値ベクトルに変換し、0.5 より大きい確率を TRUE とみなします。

実践的なインタラクティブ演習

このサンプルコードを完成させて、この演習に挑戦してみましょう。

# Extract the first model and validate 
model <- cv_models_lr$___[[___]]
validate <- cv_models_lr$___[[___]]

# Prepare binary vector of actual Attrition values in validate
validate_actual <- ___ == "Yes"

# Predict the probabilities for the observations in validate
validate_prob <- predict(___, ___, type = "response")

# Prepare binary vector of predicted Attrition values for validate
validate_predicted <- ___
コードを編集して実行