C++ API Reference for Intel® Data Analytics Acceleration Library 2020 Update 1

gbt_training_parameter.h
1 /* file: gbt_training_parameter.h */
2 /*******************************************************************************
3 * Copyright 2014-2020 Intel Corporation
4 *
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
8 *
9 * http://www.apache.org/licenses/LICENSE-2.0
10 *
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
16 *******************************************************************************/
17 
18 /*
19 //++
20 // Gradient Boosted Trees training parameter class
21 //--
22 */
23 
24 #ifndef __GBT_TRAINING_PARAMETER_H__
25 #define __GBT_TRAINING_PARAMETER_H__
26 
27 #include "algorithms/algorithm.h"
28 #include "data_management/data/numeric_table.h"
29 #include "data_management/data/data_serialize.h"
30 #include "algorithms/engines/engine.h"
31 
32 namespace daal
33 {
34 namespace algorithms
35 {
44 namespace gbt
45 {
49 namespace training
50 {
60 enum SplitMethod
61 {
62  exact = 0,
63  inexact = 1,
64  defaultSplit = inexact
65 };
66 
71 enum VariableImportanceModes
72 {
73  weight = 0x001ULL,
74  totalCover = 0x002ULL,
75  cover = 0x004ULL,
76  totalGain = 0x008ULL,
77  gain = 0x010ULL,
78 };
79 
83 namespace interface1
84 {
85 
92 /* [Parameter source code] */
93 class DAAL_EXPORT Parameter
94 {
95 public:
96  Parameter();
97 
98  SplitMethod splitMethod;
99  size_t maxIterations;
101  size_t maxTreeDepth;
102  double shrinkage;
105  double minSplitLoss;
108  double lambda;
110  double observationsPerTreeFraction;
112  size_t featuresPerNode;
115  size_t minObservationsInLeafNode;
116  bool memorySavingMode;
117  engines::EnginePtr engine;
118  size_t maxBins;
121  size_t minBinSize;
123  int internalOptions;
124 };
125 /* [Parameter source code] */
126 } // namespace interface1
127 using interface1::Parameter;
129 } // namespace training
130 }
131 }
132 } // namespace daal
133 #endif
daal::algorithms::gbt::training::exact
Definition: gbt_training_parameter.h:62
daal::algorithms::gbt::training::interface1::Parameter::observationsPerTreeFraction
double observationsPerTreeFraction
Definition: gbt_training_parameter.h:110
daal::algorithms::gbt::training::interface1::Parameter::minSplitLoss
double minSplitLoss
Definition: gbt_training_parameter.h:105
daal::algorithms::gbt::training::interface1::Parameter
Parameters for the gradient boosted trees algorithm.
Definition: gbt_training_parameter.h:93
daal::algorithms::gbt::training::interface1::Parameter::minObservationsInLeafNode
size_t minObservationsInLeafNode
Definition: gbt_training_parameter.h:115
daal::algorithms::gbt::training::interface1::Parameter::internalOptions
int internalOptions
Definition: gbt_training_parameter.h:123
daal::algorithms::gbt::training::interface1::Parameter::minBinSize
size_t minBinSize
Definition: gbt_training_parameter.h:121
daal::algorithms::gbt::training::interface1::Parameter::splitMethod
SplitMethod splitMethod
Definition: gbt_training_parameter.h:98
daal::algorithms::gbt::training::interface1::Parameter::maxTreeDepth
size_t maxTreeDepth
Definition: gbt_training_parameter.h:101
daal::algorithms::gbt::training::interface1::Parameter::lambda
double lambda
Definition: gbt_training_parameter.h:108
daal::algorithms::gbt::training::interface1::Parameter::featuresPerNode
size_t featuresPerNode
Definition: gbt_training_parameter.h:112
daal::algorithms::gbt::training::interface1::Parameter::shrinkage
double shrinkage
Definition: gbt_training_parameter.h:102
daal::algorithms::gbt::training::SplitMethod
SplitMethod
Split finding method in gradient boosted trees algorithm.
Definition: gbt_training_parameter.h:60
daal::algorithms::gbt::training::interface1::Parameter::maxBins
size_t maxBins
Definition: gbt_training_parameter.h:118
daal::algorithms::gbt::training::defaultSplit
Definition: gbt_training_parameter.h:64
daal::algorithms::gbt::training::VariableImportanceModes
VariableImportanceModes
Variable importance computation modes.
Definition: gbt_training_parameter.h:71
daal::algorithms::gbt::training::interface1::Parameter::engine
engines::EnginePtr engine
Definition: gbt_training_parameter.h:117
daal::algorithms::gbt::training::inexact
Definition: gbt_training_parameter.h:63
daal::algorithms::gbt::training::interface1::Parameter::maxIterations
size_t maxIterations
Definition: gbt_training_parameter.h:99
daal::algorithms::gbt::training::interface1::Parameter::memorySavingMode
bool memorySavingMode
Definition: gbt_training_parameter.h:116

For more complete information about compiler optimizations, see our Optimization Notice.