python利用graphviz绘制决策树

在使用python进行决策树结果可视化的时候遇到InvocationException: GraphViz’s executables not found，原因是你的电脑还没有装GraphViz这个软件。这个软件安装分两种：1、GraphViz官网下载然后自己配置环境变量；2、使用anaconda安装。本文主要讲anaconda下如何安装。

一、安装graphviz包

二、绘制决策树

from sklearn import tree #导入决策树
from sklearn.datasets import load_iris #导入datasets创建数组
iris = load_iris()#鸢尾花数据集
iris_data=iris.data #选择训练数组
iris_target=iris.target #选择对应标签数组

clf = tree.DecisionTreeClassifier() #创建决策树模型
clf=clf.fit(iris_data,iris_target) #拟合模型
import graphviz #导入决策树可视化模块
dot_data = tree.export_graphviz(clf, out_file=None) #以DOT格式导出决策树
graph = graphviz.Source(dot_data) 
graph.render(r'C:\Users\Desktop\iris') #使用garphviz将决策树转存PDF存放到桌面，文件名叫iris