Feat C++ API
A feature engineering automation tool
MyCARTreeNodeData.h
Go to the documentation of this file.
1 /*
2  * edited by William La Cava (WGL), UPenn, 2018
3  * Copyright (c) The Shogun Machine Learning Toolbox
4  * Written (w) 2014 Parijat Mazumdar
5  * All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions are met:
9  *
10  * 1. Redistributions of source code must retain the above copyright notice, this
11  * list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright notice,
13  * this list of conditions and the following disclaimer in the documentation
14  * and/or other materials provided with the distribution.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
17  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
18  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
19  * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
20  * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
21  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
22  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
23  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
25  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26  *
27  * The views and conclusions contained in the software and documentation are those
28  * of the authors and should not be interpreted as representing official policies,
29  * either expressed or implied, of the Shogun Development Team.
30  */
31 
32 
33 #ifndef MYCARTREENODEDATA_H__
34 #define MYCARTREENODEDATA_H__
35 
36 #include <shogun/lib/config.h>
37 
38 #include <shogun/multiclass/tree/TreeMachine.h>
39 #include <shogun/features/DenseFeatures.h>
40 #include <shogun/mathematics/Math.h>
41 #include <shogun/mathematics/eigen3.h>
42 
43 namespace shogun
44 {
51  {
53  int32_t attribute_id;
54 
56  SGVector<float64_t> transit_into_values;
57 
59  float64_t node_label;
60 
62  float64_t total_weight;
63 
65  float64_t weight_minus_node;
66 
69 
71  int32_t num_leaves;
72 
74  float64_t IG;
75 
78 
82  static void print_data(const MyCARTreeNodeData &data);
83  };
84 
85 
86 } /* shogun */
87 
88 #endif /* MyCARTREENODEDATA_H__ */
structure to store data of a node of CART. This can be used as a template type in TreeMachineNode cla...
SGVector< float64_t > transit_into_values
float64_t IG
WGL: IG is the Impurity Gain of this node: IG(n) = impurity(n) - impurity(l) - impurity(r)
static void print_data(const MyCARTreeNodeData &data)