1. Lära sig
    2. /
  2. Courses
    3. /
  3. HRアナリティクス：Pythonで従業員離職を予測する
Connected

exercise

GridSearch の実装

これまでに作成した max_depth と min_samples_leaf のすべての組み合わせから最適なものを見つけるために、sklearn の GridSearchCV() 関数を使いましょう。

Instruktioner

100 XP
  • GridSearchCV 関数をインポートします。
  • 先ほど定義した parameters 辞書を使って、model に GridSearchCV() を適用し、param_search として保存します。
  • 学習用データに対して param_search を学習（fit）させます。
  • 見つかった最適なパラメータを best_params_ 属性で表示します。