24 #ifndef __GBT_TRAINING_PARAMETER_H__
25 #define __GBT_TRAINING_PARAMETER_H__
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"
64 defaultSplit = inexact
71 enum VariableImportanceModes
74 totalCover = 0x002ULL,
93 class DAAL_EXPORT Parameter
98 SplitMethod splitMethod;
110 double observationsPerTreeFraction;
112 size_t featuresPerNode;
115 size_t minObservationsInLeafNode;
116 bool memorySavingMode;
117 engines::EnginePtr engine;
127 using interface1::Parameter;
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