R - rpart 예제





이번 시간엔 의사결정나무, 즉 Decision Tree인 rpart 함수를 사용한 방법을 실습합니다.


코드

#분류분석 : rpart package 사용 - 가지치기 가능
set.seed(123)
n<-sample(2,nrow(iris),replace = T, prob=c(0.7, 0.3)) #복원은 T 비복원은 F #1과 2로 나눔
n
train <-iris[n==1, ]
test <-iris[n==2, ]
dim(train); dim(test)
library(rpart)
#분류해줌
iris_rpart<-rpart(Species ~., data=train,method='class')


x11()
plot(iris_rpart)
#iris_rpart에 택스트 넣어주기
text(iris_rpart)
#가지 치기
printcp(iris_rpart)
plotcp(iris_rpart)

#데이터 값으로 볼수 있다.
cp<- iris_rpart$cptable[which.min(iris_rpart$cptable[,'xerror'])]
iris_rpart_prune <-prune(iris_rpart,cp=cp,'cp') #가지치기

plot(iris_rpart_prune)
text(iris_rpart)

#모델 평가 / 예측
rpart_pred <-predict(iris_rpart_prune,test, type='class') #가지치기 끝냄

rpart_pred
table(rpart_pred)

confusionMatrix(rpart_pred,test$Species) #0.95




샘플 데이터



본 포스팅은 KIC 캠퍼스에서 박영권 강사의 지도하에 공부하며 작성한 리포트입니다.
혹시 잘못된 내용이 있거나 문제 소지시 댓글 남겨주시면 조치하겠습니다.





'코딩 > 데이터 분석_AI' 카테고리의 다른 글

라즈베리파이에 Mariadb 설치하기 #1  (0) 2018.05.08
R - Random forest 예제  (0) 2018.05.08
R - SVM 예제  (0) 2018.05.08
R - rpart 예제  (0) 2018.05.08
R - ctree#2  (0) 2018.05.08
R - ctree #1  (0) 2018.05.08
R - 로지스틱 회귀분석 #3  (0) 2018.05.08

댓글(0)

Designed by JB FACTORY