▲ 그림 3-23 붓꽃 데이터셋에서 훈련한 결정 트리 모델

앞서 호출한 plot_tree 함수에서 filled=True로 설정하면 다수 클래스 레이블을 기준으로 노드에 색상이 지정됩니다. 그 외 다른 매개변수는 온라인 문서(https://scikit-learn.org/stable/modules/generated/sklearn.tree.plot_tree.html)를 참고하세요.

결정 트리 이미지를 보면 결정 트리가 훈련 데이터셋에서 선택한 분할을 손쉽게 분석할 수 있습니다. 각 노드의 분할 기준에 따라 왼쪽 가지는 “True”에 해당하고 오른쪽 가지는 “False”에 해당합니다.