Re: 점심시간 문제. caret을 이용한 자동튜닝을 실험하세요 ~

[12기_현지연]

library(caret)

library(C50)

library(irr)

credit <- read.csv("credit.csv", stringsAsFactors = TRUE)

str(credit)

library(caret)

set.seed(659)

in_train <- createDataPartition(credit$default, p = 0.75, list = FALSE)

credit_train <- credit[in_train, ] # 훈련 데이터 구성

credit_test <- credit[-in_train, ] # 테스트 데이터 구성

set.seed(123)

credit_model <- train( default ~ . ,  data=credit_train, method='C5.0' )

credit_model

credit_pred <- predict(credit_model, credit_test)

table(credit_pred, credit_test$default)

credit_pred  no yes

        no  141  40

        yes  34  35