我试图可视化决策树分类器的输出。这是我的密码
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')
这些颜色代表什么?我应该如何解释它们?
根据plot_tree
的文档,其填充的
参数:
filled:bool,default=False
当设置为True时,绘制节点以指示分类的多数类、回归值的极值或多输出的节点纯度。
所以在分类任务中,颜色表示每个节点上的大多数样本属于哪个类。