决策树可视化

步骤:

【1】生成决策树

from sklearn.externals.six import StringIO

savefile='E:\\test\\treepattern.dot'
with open(savefile, 'w') as f:
    f = tree.export_graphviz(clf, out_file=f, feature_names=vec.get_feature_names()) 

【2】用Graphviz软件(安装地址:http://graphviz.org/Download..php)

  1. 安装好后,记下软件dot工具位置,例如我的:D:\Program Files (x86)\Graphviz2.38\bin
  2. 打开命令提示符窗口,进入上述路径(注意:进入盘符有“:”、粘贴需要按右键不能CtrlV)

  3.进入工具位置输入 dot -Tpdf treepattern.dot -o treepattern.pdf,就能在输入的目录下找到pdf或者png了。

dot -Tpdf E:\test\treepattern.dot -o E:\test\treepattern.pdf
dot -Tpng E:\test\treepattern.dot -o E:\test\treepattern.png

 

 

 

 

 

参考资料:

http://www.cnblogs.com/zhaoxy/p/5054938.html

posted @ 2017-01-10 17:03 hqium 阅读(...) 评论(...) 编辑 收藏