코드를 실행하면 축에 나란히 놓인 전형적인 결정 트리의 결정 경계를 얻습니다.

    ▲ 그림 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 매개변수를 모두 지원합니다.

    신간 소식 구독하기
    뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.