사용자 지정 튜닝 그리드

R의 caret로 배우는 Machine Learning

Zach Mayer

Data Scientist at DataRobot and co-author of caret

사용자 지정 튜닝의 장단점

  • 사용자 지정 튜닝 그리드를 tuneGrid 인수에 전달
  • 장점
    • caret 모델 학습에 가장 유연한 방법
    • 모델 적합 방식에 완전한 제어
  • 단점
    • 모델에 대한 사전 지식 필요
    • 실행 시간이 크게 늘어날 수 있음
R의 caret로 배우는 Machine Learning

사용자 지정 튜닝 예시

# 사용자 지정 튜닝 그리드 정의
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)
R의 caret로 배우는 Machine Learning

사용자 지정 튜닝

정확도 대 무작위로 선택된 예측 변수 수의 선 그래프이며, x=10에서 정점이 있습니다.

R의 caret로 배우는 Machine Learning

연습해 봅시다!

R의 caret로 배우는 Machine Learning

Preparing Video For Download...