决策树的绘制

在使用 sklean.tree.DecisionTreeClassifier 训练决策树,并画出决策树

1.按顺序安装一下包及程序：

(1)cmd   ->  pip install GraphViz

(2)在http://www.graphviz.org/Download_windows.php下载GraphViz的zip包，解压放在某目录下 （如果你不小心关掉安装完成之后自动打开的编辑器，可以重新运行下面这个EXE：D:\Program Files\Graphviz2.38\bin\gvedit.exe）

(3)将系统路径的path后添加   D:\Program Files\Graphviz2.38\bin            

(4)pip install  pydotplus（python 2.7以上安装pydotplus 而非pydot）

(5)重启pyhton IDE

2.绘制决策树的程序

大概程序如此，不同的python版本可能会有所不同

from sklearn import tree
from IPython.display import display, Image
# from IPython.display import Image
# from sklearn.externals.six import StringIO
import pydotplus
import sys
reload(sys)
sys.setdefaultencoding('utf8')

dot_data = tree.export_graphviz(dt_clf2,out_file=None,filled=True,rounded=True,special_characters=True)  
graph = pydotplus.graph_from_dot_data(dot_data)  
display(Image(graph.create_png()))

参考博客：https://www.cnblogs.com/pinard/p/6056319.html