有没有办法让export\u graphviz像填充回归树一样填充分类树的节点?

2024-04-25 10:13:09 发布

您现在位置:Python中文网/ 问答频道 /正文

我在一个非常不平衡的数据集上使用二进制分类,当我使用export_graphviz将决策树导出到graphviz并将filled设置为True时,树的所有节点都用相同的颜色填充。有没有办法使颜色随属于另一类的可能性更高而变化?你知道吗


Tags: 数据决策树true节点颜色二进制分类export
1条回答
网友
1楼 · 发布于 2024-04-25 10:13:09

选项1:修改点输出

是否有可能(可能自动)修改点输出?一旦可以识别出有问题的节点,就可以为这些节点添加(或调整)fillcolor属性。你知道吗

选项2:sklearn中的配置

我不知道关于sklearn的细节。但从你链接的页面上看,许多设置听起来很有趣,尤其是:

  • 杂质
  • 标签
  • 类别名称

相关问题 更多 >