提问者:小点点

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


我试图可视化决策树分类器的输出。这是我的密码

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')

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


共1个答案

匿名用户

根据plot_tree的文档,其填充的参数:

filled:bool,default=False
当设置为True时,绘制节点以指示分类的多数类、回归值的极值或多输出的节点纯度。

所以在分类任务中,颜色表示每个节点上的大多数样本属于哪个类。