微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

截断决策树图中的值标签数据

如何解决截断决策树图中的值标签数据

我的决策树如下:

enter image description here



如何让我的树图如下所示:

enter image description here


我正在使用以下代码段构建我的树:
import graphviz 
from IPython.display import Image 
import pydotplus

dot_data = tree.export_graphviz(clf,out_file=None,feature_names = list(X_.columns),class_names = cnm,filled=True,rounded=True,special_characters=True,label='all',max_depth=1)  
graph = graphviz.source(dot_data)  
graph = pydotplus.graph_from_dot_data(dot_data)

这是我的点数据:
digraph Tree {
node [shape=Box,style="filled,rounded",color="black",fontname=helvetica] ;
edge [fontname=helvetica] ;
0 [label=<ProfitCent &le; 1.5<br/>gini = 0.966<br/>samples = 63644<br/>value = [1,328,5,6,6859,1,112,789,53,609,548<br/>2987,56,5632,955,8,32,323,622,33,57,28<br/>163,16,441,270,41,80,890,18,28,701<br/>2,55,153,59,242,277,288,3307,12,6<br/>57,46,14,194,9,149,43,63,295<br/>36,105,258,50,296,1940<br/>13,1481,411,29,10,9<br/>5,4410,801,795,26,45,171,88,1376<br/>33,31,34,20,628,66,132,131,2<br/>54,134,49,7,2,40,337,3<br/>24,36,625,204,142,608<br/>815,287,11,407,286,1<br/>4,62,22,172,111,23,19,4,230,3,79,52<br/>886,17,39,590,20<br/>25,25,48,146,69<br/>167,151,147,96,6<br/>101,38,64,15,116<br/>6,236,126,9<br/>11,24,35,106,102,1<br/>6,27,488,119,27<br/>11,1<br/>280,109,140<br/>17,93,21,1<br/>3,531,107,13,22<br/>9,6<br/>13,69,243,171<br/>2,47,92,65,5<br/>97,44,20<br/>2,87<br/>257,593,158,1<br/>8,10<br/>1,2<br/>3,72,1<br/>14,434,1<br/>1,509,7<br/>46,37,3<br/>1,4<br/>2,1<br/>2,180,8<br/>28,89,2<br/>1,182,52,1<br/>7,30,2<br/>6,95<br/>81,42,1<br/>170,50<br/>2,32<br/>1,75,826,164,18<br/>4,5<br/>7,2<br/>2,7<br/>3,31<br/>5,122,97,3<br/>14,1<br/>13,4<br/>9,1<br/>10,1<br/>37,1<br/>9,4<br/>1,1]<br/>class = ['6805300000']>,fillcolor="#fefcfb"] ;
1 [label=<PRDHA04 &le; 226.0<br/>gini = 0.214<br/>samples = 3831<br/>value = [1,3389,4<br/>0,0<br/>0,0<br/>6,16<br/>8,0<br/>2,216,0]<br/>class = ['6805300000']>,fillcolor="#e89251"] ;
0 -> 1 [labeldistance=2.5,labelangle=45,headlabel="True"] ;
2 [label="(...)",fillcolor="#C0C0C0"] ;
1 -> 2 ;
343 [label="(...)",fillcolor="#C0C0C0"] ;
1 -> 343 ;
436 [label=<EXTWG &le; 8.5<br/>gini = 0.971<br/>samples = 59813<br/>value = [0,327,3470,91,756,548<br/>2983,5625,954,3301,4394,793,577,77,442,2<br/>24,fillcolor="#fefdfb"] ;
0 -> 436 [labeldistance=2.5,labelangle=-45,headlabel="False"] ;
437 [label="(...)",fillcolor="#C0C0C0"] ;
436 -> 437 ;
3560 [label="(...)",fillcolor="#C0C0C0"] ;
436 -> 3560 ;
}

如果我的数据集很大,那么自动标签数据会被截断,但我也想要更少的数据量。
提前致谢

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。