【发布时间】:2021-02-10 03:53:31
【问题描述】:
我创建了一个graphviz 图来可视化决策树。现在我想在所有边的图形上添加标签(例如True、False)。
我使用正则表达式来操作.dot 文件,手动添加标签True 和False。产生以下文件
digraph Tree {
node [shape=box, style="rounded", color="black", fontname=helvetica] ;
edge [fontname=helvetica] ;
0 [label="s5 smaller
or equal to 0.0"] ;
1 [label="bmi smaller
or equal to 0.0"] ;
0 -> 1 [labeldistance=2.5, labelangle=45, headlabel="True"] ;
3 [label="value = 110.6"] ;
1 -> 3 [headlabel="True"] ;
4 [label="value = 161.0"] ;
1 -> 4 [headlabel="False"] ;
2 [label="bmi smaller
or equal to 0.0"] ;
0 -> 2 [labeldistance=2.5, labelangle=-45, headlabel="False"] ;
5 [label="value = 174.0"] ;
2 -> 5 [headlabel="True"] ;
6 [label="value = 237.7"] ;
2 -> 6 [headlabel="False"] ;
}
这会产生以下情节:
但是,标签没有很好地排列(箭头切断了标签),我想相应地格式化文本。有没有办法在graphviz 的地块中做到这一点?
谢谢!
【问题讨论】:
标签: python-3.x plot graphviz dot