aboutsummaryrefslogtreecommitdiffstats
path: root/searchlib/src/test/files/gbdt_set_inclusion_test.xml
blob: 80892a1dda13d6b55135a0b3ef18e48f59baa64d (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
<?xml version="1.0" encoding="UTF-8"?>
<!-- Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. -->
<MlrFunction name="CHANGE THIS NUMBER" featuredef="mlrfeaturedefs.xml" version="1.0">

  <!-- ADD SCORE STANDARDIZATION OR CALIBRATION HERE -->

    <DecisionTree loss="logistic">

    <!-- ADD EARLY EXIT HERE -->

        <Forest>
            <Tree id="0">
                 <Node feature="AGE_GROUP$" value="2" id="N0_1">
                     <Node feature="EDUCATION_LEVEL$" value="0" id="N0_2">
                         <Response value="-0.2500000" id="T0_1"/>
                         <Response value="0.1250000" id="T0_2"/>
                     </Node>
                     <Node feature="AGE_GROUP$" value="1" id="N0_3">
                         <Response value="0.1250000" id="T0_3"/>
                         <Response value="0.2500000" id="T0_4"/>
                     </Node>
                 </Node>
            </Tree>
            <Tree id="1">
                 <Node feature="AGE_GROUP$" value="2" id="N1_1">
                     <Node feature="EDUCATION_LEVEL$" value="0" id="N1_2">
                         <Response value="-0.2189117" id="T1_1"/>
                         <Response value="-0.0000000" id="T1_2"/>
                     </Node>
                     <Node feature="EDUCATION_LEVEL$" value="0" id="N1_3">
                         <Response value="0.1094559" id="T1_3"/>
                         <Response value="0.2343953" id="T1_4"/>
                     </Node>
                 </Node>
            </Tree>
            <Tree id="2">
                 <Node feature="AGE_GROUP$" value="2" id="N2_1">
                     <Response value="-0.0962185" id="T2_1"/>
                     <Node feature="EDUCATION_LEVEL$" value="0" id="N2_2">
                         <Node feature="AGE_GROUP$" value="1" id="N2_3">
                             <Response value="0.0000000" id="T2_2"/>
                             <Response value="0.2055456" id="T2_3"/>
                         </Node>
                         <Response value="0.2055530" id="T2_4"/>
                     </Node>
                 </Node>
            </Tree>
            <Tree id="3">
                 <Node feature="EDUCATION_LEVEL$" value="0" id="N3_1">
                     <Response value="0.0905977" id="T3_1"/>
                     <Response value="0.1812016" id="T3_2"/>
                 </Node>
            </Tree>
            <Tree id="4">
                 <Node feature="EDUCATION_LEVEL$" value="0,1" id="N4_1">
                     <Node feature="AGE_GROUP$" value="2" id="N4_2">
                         <Node feature="EDUCATION_LEVEL$" value="0" id="N4_3">
                             <Response value="-0.1917720" id="T4_1"/>
                             <Response value="-0.0000000" id="T4_2"/>
                         </Node>
                         <Node feature="AGE_GROUP$" value="1" id="N4_4">
                             <Node feature="EDUCATION_LEVEL$" value="0" id="N4_5">
                                 <Response value="0.0000000" id="T4_3"/>
                                 <Response value="0.1608304" id="T4_4"/>
                             </Node>
                             <Response value="0.1708644" id="T4_5"/>
                         </Node>
                     </Node>
                     <Response value="0.1923393" id="T4_6"/>
                 </Node>
            </Tree>
            <Tree id="5">
                 <Node feature="EDUCATION_LEVEL$" value="foo,bar" id="N5_1">
                     <Node feature="AGE_GROUP$" value="2" id="N5_2">
                         <Node feature="EDUCATION_LEVEL$" value="baz" id="N5_3">
                             <Response value="-0.1696624" id="T5_1"/>
                             <Response value="-0.0000000" id="T5_2"/>
                         </Node>
                         <Node feature="AGE_GROUP$" value="1" id="N5_4">
                             <Node feature="EDUCATION_LEVEL$" value="0" id="N5_5">
                                 <Response value="0.0000000" id="T5_3"/>
                                 <Response value="0.1438091" id="T5_4"/>
                             </Node>
                             <Response value="0.1521967" id="T5_5"/>
                         </Node>
                     </Node>
                     <Response value="0.2003772" id="T5_6"/>
                 </Node>
            </Tree>
            <Tree id="6">
                 <Response value="-0.0108278" id="T6_1"/>
            </Tree>
            <Tree id="7">
                 <Node feature="EDUCATION_LEVEL$" value="0" id="N7_1">
                     <Response value="-0.1500528" id="T7_1"/>
                     <Node feature="GENDER$" value="1" id="N7_2">
                         <Response value="0.0652894" id="T7_2"/>
                         <Response value="0.1543407" id="T7_3"/>
                     </Node>
                 </Node>
            </Tree>
            <Tree id="8">
                 <Node feature="AGE_GROUP$" value="1" id="N8_1">
                     <Response value="0.0000000" id="T8_1"/>
                     <Response value="0.1569706" id="T8_2"/>
                 </Node>
            </Tree>
            <Tree id="9">
                 <Node feature="AGE_GROUP$" value="1" id="N9_1">
                     <Response value="0.0000000" id="T9_1"/>
                     <Node feature="EDUCATION_LEVEL$" value="1" id="N9_2">
                         <Response value="0.0000000" id="T9_2"/>
                         <Response value="0.1405829" id="T9_3"/>
                     </Node>
                 </Node>
            </Tree>
        </Forest>
    </DecisionTree>
</MlrFunction>