Lưới tinh chỉnh tùy chỉnh

Machine Learning với caret trong R

Zach Mayer

Data Scientist at DataRobot and co-author of caret

Ưu và nhược điểm của tinh chỉnh tùy chỉnh

Truyền lưới tinh chỉnh tùy chỉnh vào đối số tuneGrid
Ưu điểm
- Cách linh hoạt nhất để huấn luyện mô hình caret
- Kiểm soát hoàn toàn cách mô hình được huấn luyện
Nhược điểm
- Cần có hiểu biết về mô hình
- Có thể làm tăng mạnh thời gian chạy

Ví dụ tinh chỉnh tùy chỉnh

# 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)

Tinh chỉnh tùy chỉnh

Hình cho thấy biểu đồ đường giữa độ chính xác và số lượng predictor chọn ngẫu nhiên, đạt đỉnh tại x bằng 10.

Hãy thực hành!

Machine Learning với caret trong R