カスタムチューニングでランダムフォレストを学習する
train() 関数が提供するデフォルトのチューニンググリッドを確認したので、ここからはモデルをもう少しカスタマイズしていきます。
mtry には、2 からデータセットの列数まで任意の数の値を指定できます。実務では、mtry を大きくしすぎても効果は逓減することが多いので、ここではシンプルな 2 つのモデル(mtry = 2 と mtry = 3)に加えて、もう少し複雑なモデル(mtry = 7)も試せるカスタムチューニンググリッドを使います。
この演習はコースの一部です
Rで学ぶ caret を使った Machine Learning
実践的なインタラクティブ演習
このサンプルコードを完成させて、この演習に挑戦してみましょう。
# Define the tuning grid: tuneGrid
tuneGrid <- data.frame(
.mtry = ___,
.splitrule = "___",
.min.node.size = ___
)