1. 学ぶ
    2. /
  2. コース
    3. /
  3. Rで学ぶTree-Based ModelsによるMachine Learning
Connected

演習

ROC曲線をプロットする

ブーステッドツリーが最も高いAUCを出すことを再確認しました。数値も大事ですが、可視化するとさらにわかりやすくなります。これらの結果を可視化しましょう。

ここでは、すべてのROC曲線を1つの共通プロットに重ねて、モデルの性能を比較します。AUCはROC曲線の下の面積そのものなので、ブーステッドモデルの曲線が最も大きな面積を持ち、プロットの左上に近い位置にくるはずです。

予測のtibbleであるpreds_combinedは既に読み込まれています。

指示

100 XP
  • preds_combinedのtibbleを整形し、"preds_"で始まるすべての列を列から行に変換します。元の列名は"model"列に、値は"predictions"という列に入れます。
  • 結果をmodelでグループ化します。
  • すべてのカットオフについてROCの値を計算します。
  • 曲線をグラフィカルにプロットします。