1. 学ぶ
  2. /
  3. コース
  4. /
  5. XGBoost で学ぶ極限の勾配ブースティング

Connected

演習

AUC を測定する

これまでに、クロスバリデーションを使って(エラーから変換して)平均のアウトオブサンプル精度を計算しました。ほかの評価指標を計算するのもとても簡単です。xgb.cv() の metrics パラメータに、その指標(または指標のリスト)を引数として渡すだけです。

この演習では、二値分類でよく使われるもう一つの指標、曲線下面積("auc")を計算します。これまでと同様に、作業スペースには churn_data、DMatrix の churn_dmatrix、パラメータ辞書の params が用意されています。

指示

100 XP
  • 5 回のブースティングと "auc" を指標として、3 分割のクロスバリデーションを実行します。
  • cv_results の "test-auc-mean" 列を出力します。