Власні сітки налаштування

Machine Learning з пакетом caret в R

Zach Mayer

Data Scientist at DataRobot and co-author of caret

Плюси й мінуси власного налаштування

Передайте власні сітки налаштування в аргумент tuneGrid
Переваги
- Найгнучкіший спосіб навчання моделей у caret
- Повний контроль над тим, як навчається модель
Недоліки
- Потрібні базові знання моделі
- Може суттєво збільшити час виконання

Приклад власного налаштування

# Define a custom tuning grid
myGrid <- data.frame(mtry = c(2, 3, 4, 5, 10, 20))

# Fit a model with a custom tuning grid
set.seed(42)
model <- train(
  Class ~ ., 
  data = Sonar, 
  method = "ranger",            
  tuneGrid = myGrid
)

# Plot the results
plot(model)

Власне налаштування

Зображення: лінійний графік точності залежно від випадково вибраних предикторів, із піком при x = 10.

Давайте потренуємось!

Machine Learning з пакетом caret в R