决策树分类器 sklearn 中节点的不同颜色表示什么？

Question

我正在尝试可视化决策树分类器的输出。这是我的代码

from sklearn import tree
import matplotlib.pyplot as plt

plt.figure(figsize=(30,15))
tree.plot_tree(dt2,filled=True,fontsize=8)
plt.show()
plt.savefig('dtree.jpg')

这是我得到的图像。

这些颜色代表什么？我应该如何解释它们？

Answer 1

根据 plot_tree 为其 filled 参数的文档：

<块引用>

填充：bool，默认=False
设置为 True 时，绘制节点以指示用于分类的多数类、用于回归的极端值或用于多输出的节点纯度。

因此在分类任务中，颜色表示每个节点的大多数样本属于哪个类。