코드를 실행하면 축에 나란히 놓인 전형적인 결정 트리의 결정 경계를 얻습니다.
▲ 그림 3-20 결정 트리가 학습한 붓꽃 데이터셋의 결정 경계
사이킷런에는 다음 코드처럼 훈련이 끝난 결정 트리 모델을 손쉽게 시각화할 수 있는 멋진 기능이 있습니다.29
>>> from sklearn import tree
>>> tree.plot_tree(tree_model)
>>> plt.show()
▲ 그림 3-21 plot_tree( ) 함수로 그린 결정 트리
29 역주 plot_tree 함수를 사용하면 .dot 파일을 만들지 않고 그림 3-24와 비슷한 트리를 바로 그릴 수 있습니다. 이 함수는 export_graphviz 함수에서 사용할 수 있는 class_names, feature_names, rounded, filled 매개변수를 모두 지원합니다.