决策树可视化

使用pydotplus和graphviz对决策树进行可视化。

安装包

安装包的方法可以参考这篇帖子。

打印决策树

Python

# 打印预测用户打算在什么平台申领的决策树
from sklearn.tree import export_graphviz

dot_data = export_graphviz(
    clf,
    out_file=None,
    feature_names=["性别", "年龄", "XX"],
    class_names=clf.classes_,
    filled=True,
    rounded=True,
    special_characters=True,
)

import pydotplus

graph = pydotplus.graph_from_dot_data(dot_data)

# 添加环境变量，否则会报错找不到 Graphviz 的安装路径
import os

os.environ["PATH"] += os.pathsep + "C:/Program Files/Graphviz/bin"

# 导出为 svg 格式的图片，这样能够支持中文
graph.write_svg("文件名.svg")

如果报错找不到 Graphviz 的安装路径，记得把 Graphviz 的 bin 目录路径通过os.environ["PATH"] += os.pathsep + 'C:/Program Files/Graphviz/bin'添加到环境变量中。

导出结果

将决策树可视化可以帮助我们理解这颗决策树究竟是如何运行的。

评论