Python决策树怎么画_分类树与回归树模型训练及Graphviz结构可视化

sklearn.tree.plot_tree是最轻量的决策树可视化方案，需传入已训练模型及feature_names等参数；分类树value为各类样本数，回归树value为预测值；export_graphviz需系统级graphviz支持且注意precision等参数。

怎么用 plot_tree 快速画出决策树结构图

不用 Graphviz 也能看树，sklearn.tree.plot_tree 是最轻量的方案。它直接在 matplotlib 里渲染，适合快速检查分裂逻辑、深度、样本数分布。

常见错误是传入未训练的模型——plot_tree 要求模型已调用 fit()；另外默认不显示特征名和类别名，图看起来像天书。

• 必须传入已训练的 DecisionTreeClassifier 或 DecisionTreeRegressor
• 加 feature_names=feature_names 和 class_names=class_names（分类树才需要）才能看懂节点含义
• 小树用 max_depth=3 控制显示层级，否则深层树会挤成一团
• 回归树的 class_names 参数无效，别传；它的叶节点显示的是预测值（value）而非类别

from sklearn.tree import plot_tree
import matplotlib.pyplot as plt

plt.figure(figsize=(10, 6))
plot_tree(clf, feature_names=iris.feature_names, class_names=iris.target_names, filled=True, rounded=True, fontsize=10)
plt.show()

Graphviz 导出 PDF/SVG 需要绕过哪些环境坑

想导出高清矢量图（比如插论文），就得走 export_graphviz + graphviz 工具链。真正卡住人的从来不是 Python 代码，而是系统级依赖。

典型现象：跑 dot -V 报错 command not found，或 Python 里提示 ExecutableNotFound: failed to execute ['dot']。

立即学习“Python免费学习笔记（深入）”；

人民网AIGC-X

国内科研机构联合推出的AI生成内容检测工具

下载

• Mac 用户装了 graphviz Python 包，但没装系统级 graphviz 二进制（brew install graphviz 缺一不可）
• Windows 用户装完 Graphviz 安装包后，必须手动把 dot.exe 所在路径（如 C:\Program Files\Graphviz2.44\bin）加进系统 PATH
• Conda 环境下慎用 conda install python-graphviz，它不装 dot，推荐 conda install -c conda-forge python-graphviz graphviz
• export_graphviz 输出的是 DOT 文本，得用 Source（来自 graphviz 包）封装后才能 render，不能直接写文件再双击打开

分类树和回归树在可视化时的关键差异

同一个 plot_tree 函数，画分类树和回归树时表现不同，不是 bug，是设计使然。忽略这点会导致误读叶节点数值。

比如你看到一个叶节点标着 value = [12.3]，这在分类树里毫无意义（该是数组计数），但在回归树里就是预测值。

• 分类树的 value 是各类别样本数量数组，如 [5, 0, 17] 表示该叶有 5 个 class 0、0 个 class 1、17 个 class 2
• 回归树的 value 是标量数组，如 [14.2]，即该叶所有样本目标值的均值（也是预测输出）
• filled=True 下，分类树按类别分布上色，回归树按预测值大小上色（颜色深浅对应数值高低）
• 回归树不支持 class_names，传了会静默忽略；分类树若没传 class_names，只显示数字索引

为什么 export_graphviz 的 precision 和 rounded 很关键

默认导出的 DOT 文件里，分裂阈值和 impurity 值全是长浮点数（如 0.49999999999999994），不仅难读，还可能因精度问题导致 Graphviz 渲染失败或报 warning。

这不是显示问题，是底层解析容错性问题——某些旧版 dot 对超长小数处理不稳定。

• precision=2 把所有数字截到小数点后两位，既干净又不影响理解分裂逻辑
• rounded=True 让 Graphviz 自动对数字加千分位逗号和四舍五入，比纯 truncating 更稳妥
• 别依赖 proportion=True 来“美化” value 显示，它把计数转成比例，反而让调试样本分布变得更麻烦
• 如果树很深，务必加 max_depth=3，否则生成的 DOT 文件可能超百 MB，dot 渲染直接卡死