カスタムチューニンググリッド

Rで学ぶ caret を使った Machine Learning

Zach Mayer

Data Scientist at DataRobot and co-author of caret

カスタムチューニングの長所と短所

# カスタムチューニンググリッドを定義
myGrid <- data.frame(mtry = c(2, 3, 4, 5, 10, 20))

# カスタムチューニンググリッドでモデルを学習
set.seed(42)
model <- train(
  Class ~ ., 
  data = Sonar, 
  method = "ranger",            
  tuneGrid = myGrid
)

# 結果をプロット
plot(model)

この画像は、精度とランダムに選択された予測変数数の関係を示す折れ線グラフで、x=10でピークがあります。

Rで学ぶ caret を使った Machine Learning