python 출력 결정 트 리 도형

windows10：
1. 먼저 pip 에 pydotplus 와 graphviz 를 설치 해 야 합 니 다.

pip install pydotplus
pip install graphviz

2, www. graphviz. org msi 파일 을 다운로드 하고 설치 합 니 다.3. 시스템 환경 변수 path 에 두 가 지 를 추가 합 니 다.

C:\Program Files (x86)\Graphviz2.38\bin
C:\Program Files (x86)\Graphviz2.38
#  graphviz          。

4, python 에서 사용 하 는 방법:

from sklearn.externals.six import StringIO  
import pydotplus

#drt DecisionTreeClassifier()，    fit             。
dot_data = StringIO()  
tree.export_graphviz(drt, out_file=dot_data)   
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())   
graph.write_png("out.png")  #       out.png

#         pdf：
dot_data = tree.export_graphviz(drt, out_file=None) 
graph = graphviz.Source(dot_data) 
graph.render()