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

Connected

演習

max_depth のチューニング

この演習では、各ブースティングラウンドで生成される木の最大深さを決めるパラメータである max_depth をチューニングします。値が小さければ浅い木に、値が大きければ深い木になります。

指示

100 XP
  • max_depths というリストを作成し、次の "max_depth" の値を格納してください: 2、5、10、20。
  • for ループを使って max_depths リストを反復処理します。
  • for ループの各イテレーションで "max_depth" を系統的に変化させ、早期終了(5 ラウンド)付きの 2 分割交差検証、ブースティングラウンド数 10、評価指標 "rmse"、seed は 123 で実行してください。出力は DataFrame になるようにしてください。