决策树可视化¶
使用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'添加到环境变量中。
导出结果¶
将决策树可视化可以帮助我们理解这颗决策树究竟是如何运行的。