Python + Graphviz 可视化 决策树

1. 环境

1. 纯python，3.6.8
2. Windows10 x64系统

2. 安装Graphviz

2.1 安装软件

进入官方网站下载http://www.graphviz.org/download/，找到windows稳定版本的安装包

点进去后一路选择，我选择的是cmake的Release 2.44.1版本

安装的时候，记得选中将安装路径添加到path中。

打开cmd，运行dot -v，如果显示版本则说明安装成功。

2.2 安装库

参考https://www.mlln.cn/2018/09/25/python-graphviz%E5%9B%BE%E6%95%B0%E6%8D%AE%E5%8F%AF%E8%A7%86%E5%8C%96%E5%85%A5%E9%97%A8/#undefined

graphviz为Graphviz图形绘制软件提供了一个简单的纯Python接口。要使用pip安装它，请运行以下命令：

pip install graphviz

2.3 最重要的一步！！！

参考https://blog.csdn.net/qq_42479987/article/details/109403618

在命令行输入

dot -c

在命令行再输入

dot -v

如果出现以下字段，则说明能够使用Graphviz进行可视化了。

3. 运行代码，进行可视化

参考https://sklearn.apachecn.org/docs/master/11.html
这里就是将文档中的代码一行行粘贴了过来，完整代码如下

>>> from sklearn import tree
>>> X = [[0, 0], [1, 1]]
>>> Y = [0, 1]
>>> clf = tree.DecisionTreeClassifier()
>>> clf = clf.fit(X, Y)
>>> clf.predict([[2., 2.]])
>>> clf.predict_proba([[2., 2.]])
>>> from sklearn.datasets import load_iris
>>> from sklearn import tree
>>> iris = load_iris()
>>> clf = tree.DecisionTreeClassifier()
>>> clf = clf.fit(iris.data, iris.target)
>>> import graphviz
>>> dot_data = tree.export_graphviz(clf, out_file=None)
>>> graph = graphviz.Source(dot_data)
>>> graph.render("iris")
>>> dot_data = tree.export_graphviz(clf, out_file=None,
...                      feature_names=iris.feature_names,  
...                      class_names=iris.target_names,  
...                      filled=True, rounded=True,  
...                      special_characters=True)  
>>> graph = graphviz.Source(dot_data)  
>>> graph

如果按照上面一步步来的话，最终应该能够顺利出图。