trainControl 재사용

R의 caret로 배우는 Machine Learning

Max Kuhn

Software Engineer at RStudio and creator of caret

실전 예시

데이터: 통신사 고객 이탈
여러 모델을 학습해 최적 모델 선택
모든 모델에 동일한 학습/테스트 분할 사용
공용 trainControl 객체 생성

예시: 고객 이탈 데이터

# 타깃 변수 요약
library(caret)
library(C50)
data(churn)
table(churnTrain$churn) / nrow(churnTrain)

      yes        no 
0.1449145 0.8550855

예시: 고객 이탈 데이터

# 학습/테스트 인덱스 생성
set.seed(42)
myFolds <- createFolds(churnTrain$churn, k = 5)

# 클래스 분포 비교
i <- myFolds$Fold1
table(churnTrain$churn[i]) / length(i)

      yes        no 
0.1441441 0.8558559

예시: 고객 이탈 데이터

myControl <- trainControl(
  summaryFunction = twoClassSummary,
  classProbs = TRUE,
  verboseIter = TRUE,
  savePredictions = TRUE,
  index = myFolds
)

폴드를 활용해 trainControl 객체 생성
모든 모델에 동일한 교차검증 폴드 적용

실습해 봅시다!

R의 caret로 배우는 Machine Learning