하지만 사이킷런의 결정 트리를 그리기 위해 GraphViz 프로그램을 사용하면 더 나은 그림을 얻을 수 있습니다. 사이킷런은 훈련한 후 결정 트리를 .dot 파일로 추출할 수 있는 멋진 기능을 가지고 있습니다. GraphViz 프로그램을 사용하여 이 파일을 시각화할 수 있습니다. 이 프로그램은 http://www.graphviz.org/download/에서 무료로 내려받을 수 있고 리눅스, 윈도, macOS를 지원합니다. GraphViz 외에도 PyDotPlus 파이썬 라이브러리를 사용하겠습니다. GraphViz와 비슷한 기능을 가지고 있고 .dot 데이터 파일을 결정 트리 이미지 파일로 변환합니다. 웹 사이트에서 GraphViz를 내려받아 설치한 후(http://www.graphviz.org/Download.php에서 안내문을 내려받아 참고) pip 패키지 관리자를 사용하여 PyDotPlus를 바로 설치할 수 있습니다.
터미널 명령줄에서 다음 명령을 실행하세요.
> pip3 install pydotplus
또는 아나콘다에서 다음 명령을 실행합니다.
> conda install pydotplus
Note ≡ PyDotPlus 필요 라이브러리 설치
일부 시스템에서는 PyDotPlus에서 필요한 패키지를 수동으로 설치해야 합니다. 다음 명령을 참고하세요.
> pip3 install graphviz
> pip3 install pyparsing
또는 아나콘다에서 다음 명령을 실행합니다.
> conda install graphviz
> conda install pyparsing