标签： python scikit-learn graphviz decision-tree
我正在对一个非常不平衡的数据集进行二进制分类，当我使用export_graphviz将决策树导出到graphviz并将filled设置为True时，树的所有节点都被填充具有相同的颜色。有没有办法使颜色随着属于替代类的可能性更高而变化？
filled
True
答案 0 :(得分：0)
（可能自动）修改点输出是否可能？一旦确定了有问题的节点，就可以为这些节点添加（或改编）fillcolor属性。
我不知道有关sklearn的详细信息。但是在您链接的页面上，许多设置听起来很有趣，尤其是这些：