Question

我的决策树如下：

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

我正在使用以下代码段构建我的树：

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

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

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

0 个答案: