【问题标题】:Is there a way to make export_graphviz fill nodes of classification trees as if it were a regression tree?有没有办法让 export_graphviz 填充分类树的节点,就好像它是回归树一样?
【发布时间】:2019-08-07 18:41:59
【问题描述】:

我正在对一个非常不平衡的数据集进行二进制分类,当我使用export_graphviz 将我的决策树导出到graphviz 并将filled 设置为True 时,树的所有节点都填充了相同的颜色。有没有办法让颜色随着属于替代类别的概率更高而变化?

【问题讨论】:

    标签: python scikit-learn graphviz decision-tree


    【解决方案1】:

    选项1:修改点输出

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

    选项 2:在 sklearn 中配置

    我不知道有关 sklearn 的详细信息。但是从您链接的页面来看,许多设置听起来很有趣,尤其是这些:

    • 杂质
    • 标签
    • 类名

    【讨论】:

      猜你喜欢
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 2018-10-31
      • 1970-01-01
      • 2021-12-04
      • 1970-01-01
      • 2011-07-17
      • 1970-01-01
      相关资源
      最近更新 更多