1. 学ぶ
  2. /
  3. コース
  4. /
  5. Tidyverse で学ぶ Machine Learning

Connected

演習

単一モデルの予測

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

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

指示

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