1. 学ぶ
  2. /
  3. コース
  4. /
  5. Rで学ぶ caret を使った Machine Learning

Connected

演習

カスタムチューニングでランダムフォレストを学習する

train() 関数が提供するデフォルトのチューニンググリッドを確認したので、ここからはモデルをもう少しカスタマイズしていきます。

mtry には、2 からデータセットの列数まで任意の数の値を指定できます。実務では、mtry を大きくしすぎても効果は逓減することが多いので、ここではシンプルな 2 つのモデル(mtry = 2 と mtry = 3)に加えて、もう少し複雑なモデル(mtry = 7)も試せるカスタムチューニンググリッドを使います。

指示1 / 2

undefined XP
    1
    2
  • カスタムチューニンググリッドを定義します。
    • 各ノードで分割に候補とする変数数 .mtry を、2、3、7 のベクトルに設定します。
    • 分割ルール .splitrule を "variance" に設定します。
    • 最小ノードサイズ .min.node.size を 5 に設定します。