相关疑难解决方法(0)

我正在使用scikit的回归树函数和graphviz来生成一些决策树的精彩,易于解释的视觉效果:

dot_data = tree.export_graphviz(Run.reg, out_file=None, 
                         feature_names=Xvar,  
                         filled=True, rounded=True,  
                         special_characters=True) 
graph = pydotplus.graph_from_dot_data(dot_data)
graph.write_png('CART.png')
graph.write_svg("CART.svg")

这完全运行,但我想改变颜色方案,如果可能的话？该图表示CO ₂通量,因此我想将负值设为绿色且为正棕色.我可以导出为svg而不是手动更改所有内容,但是当我这样做时,文本与盒子不完全对齐,因此手动更改颜色并修复所有文本为我的工作流程添加了一个非常繁琐的步骤,我真的很喜欢避免!

此外,我已经看到一些树,其中连接节点的线的长度与分裂所解释的%方差成比例.如果可能的话,我也希望能够做到这一点？