from sklearn.externals.six import StringIO
dot_data1 = StringIO()
tree.export_graphviz(clf2, out_file=dot_data1)
graph1 = pydotplus.graph_from_dot_data(dot_data1.getvalue())
graph1.write_pdf("tree1.pdf")
运行时出现pydotplus.graphviz.InvocationException: GraphViz's executables not found 错误
用brew命令安装graphviz,过程中会自动安装graphviz的依赖包:
如果未安装brew,会提示错误brew command not found,查看解决办法:mac安装brew(brew command not found错误解决方法)
brew install graphviz
(如果之前已经安装过,可用brew reinstall graphviz命令重新安装)
过程中会安装依赖包:Installing dependencies for graphviz: libpng, freetype, fontconfig, jpeg, libtiff, webp, gd and libtool
成功安装后,则可使用以下命令绘制决策树的可视化图:
import pydotplus
dot_data1 = tree.export_graphviz(clf2, out_file=None)
graph1 = pydotplus.graph_from_dot_data(dot_data1)
graph1.write_pdf("tree1.pdf")
参考:https://blog.csdn.net/coder_Gray/article/details/81632003