【发布时间】:2019-08-07 18:41:59
【问题描述】:
我正在对一个非常不平衡的数据集进行二进制分类,当我使用export_graphviz 将我的决策树导出到graphviz 并将filled 设置为True 时,树的所有节点都填充了相同的颜色。有没有办法让颜色随着属于替代类别的概率更高而变化?
【问题讨论】:
标签: python scikit-learn graphviz decision-tree
我正在对一个非常不平衡的数据集进行二进制分类,当我使用export_graphviz 将我的决策树导出到graphviz 并将filled 设置为True 时,树的所有节点都填充了相同的颜色。有没有办法让颜色随着属于替代类别的概率更高而变化?
【问题讨论】:
标签: python scikit-learn graphviz decision-tree
是否有可能(可能自动)修改点输出?一旦您可以识别有问题的节点,您就可以为这些节点添加(或调整)fillcolor 属性。
我不知道有关 sklearn 的详细信息。但是从您链接的页面来看,许多设置听起来很有趣,尤其是这些:
【讨论】: