decision_tree 의사 결정 나무가 그려졌습니다. sales_sum < 55.5를 설명하면 다음과 같습니다. sales_sum < 55.5를 기준으로 N으로 나뉘는 노드(node)(분기)와 sales_sum < 173로 다시 나누는 노드가 있다는 의미입니다. 그려진 그림은 decision_tree 결과를 시각적으로 확인할 수 있어 좀 더 알아보기 쉽지만, 그림이 투박하고 정보량이 풍부하지 못해 여전히 이해하기가 쉽지 않습니다. rattle 패키지를 이용해서 decision_tree 모델을 좀 더 깔끔하고 직관적으로 그려 보겠습니다. rattle 패키지를 설치하세요.
install.packages("rattle") # 패키지 설치 WARNING: Rtools is required to build R packages but is not currently installed. Please download and install the appropriate version of Rtools before proceeding: ... 생략 ... library(rattle) # 패키지 로딩
그래프를 그릴 때는 rattle 패키지의 fancyRpartPlot() 함수를 사용합니다.
fancyRpartPlot(decision_tree) # 의사 결정 나무 깔끔하게 그리기