可视化决策树

1.代码

image.png

首先，我在网上搜到了这样的代码

2.安装python库

image.png

3.问题和解决

运行提示错误

image.png
百度了好多，只有一个说的比较好，解决了我的问题。
http://wenda.chinahadoop.cn/question/5074

#  没有安装GraphViz‘s executables.我是用pip安装的Graphviz,但是Graphviz不是一个python tool，你仍然需要安装GraphViz‘s executables.
#  将GraphViz安装目录的bin目录放到环境变量的path路径中
#  on ubuntu 14.04     sudo apt-get install graphviz
#  windows下：
import os
os.environ["PATH"] += os.pathsep + 'C:/Program Files (x86)/Graphviz2.38/bin/'

这是一个安装方法
https://blog.csdn.net/tina_ttl/article/details/51778551#安装graphviz
官网
http://www.graphviz.org/

image.png
下载，我下载的zip版本，然后解压缩，把bin文件所在的目录，添加到系统环境变量的path里面。
然后在代码里添加一句

import os
os.environ["PATH"] += os.pathsep + 'D:/graphviz/release/bin'

完整代码

from sklearn.externals.six import StringIO

from sklearn.datasets import load_iris
from sklearn import tree

import pydotplus

import os
os.environ["PATH"] += os.pathsep + 'D:/graphviz/release/bin'


iris = load_iris()
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)

dot_data = StringIO()
tree.export_graphviz(clf, out_file=dot_data)
graph = pydotplus.graph_from_dot_data(dot_data.getvalue())
graph.write_pdf("iris.pdf")

image.png

完成啦，已经生成了。

image.png

如图