步骤:
【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)
- 安装好后,记下软件dot工具位置,例如我的:D:Program Files (x86)Graphviz2.38in
- 打开命令提示符窗口,进入上述路径(注意:进入盘符有“:”、粘贴需要按右键不能CtrlV)
3.进入工具位置输入 dot -Tpdf treepattern.dot -o treepattern.pdf,就能在输入的目录下找到pdf或者png了。
dot -Tpdf E: est reepattern.dot -o E: est reepattern.pdf
dot -Tpng E: est reepattern.dot -o E: est reepattern.png
参考资料:
http://www.cnblogs.com/zhaoxy/p/5054938.html