决策树的可视化

1. 首先安装 graphviz 并设置环境变量, 很多文章都写这个教程了, 这里就不写了

2. 安装 python-graphviz

conda  install python-graph-viz

3. 生成图片文件

import graphviz
from sklearn.tree import DecisionTreeClassifier,export_graphviz
from sklearn.datasets import load_iris

iris = load_iris()
X,y = iris.data,iris.target
tree = DecisionTreeClassifier(max_depth=3)
tree.fit(X,y)

# out_file不为None, 就不会返回值, 而是会创建out_file指定的文本文件
export_graphviz(tree,out_file="iris",feature_names=iris.feature_names,class_names=iris.target_names,filled=True,rounded=True)
# 此时可以在命令行窗口执行命令行执行, 生成需要的格式文件
# dot -Tpng iris -o tree.png 命令生成png图片, 其中iris是指定刚才生成的文件名, tree.png是指生成的图片名称
# dot -Tps iris -o tree.ps 可以生成ps文件, 运行又可以生成pdf文件

即可看到该文件