48 #ifndef __DECISION_FOREST_TRAINING_PARAMETER_H__
49 #define __DECISION_FOREST_TRAINING_PARAMETER_H__
51 #include "algorithms/algorithm.h"
52 #include "data_management/data/numeric_table.h"
53 #include "data_management/data/data_serialize.h"
54 #include "services/daal_defines.h"
55 #include "algorithms/engines/mt19937/mt19937.h"
69 namespace decision_forest
84 enum VariableImportanceMode
103 enum ResultToComputeId
105 computeOutOfBagError = 0x00000001ULL
121 class DAAL_EXPORT Parameter
124 Parameter() : nTrees(100),
125 observationsPerTreeFraction(1.),
126 featuresPerNode(0), maxTreeDepth(0),
127 minObservationsInLeafNode(0),
129 impurityThreshold(0.),
132 memorySavingMode(
false),
134 engine(engines::mt19937::Batch<>::create()) {}
137 double observationsPerTreeFraction;
139 size_t featuresPerNode;
143 size_t minObservationsInLeafNode;
146 engines::EnginePtr engine;
147 double impurityThreshold;
149 VariableImportanceMode varImportance;
150 DAAL_UINT64 resultsToCompute;
151 bool memorySavingMode;
156 using interface1::Parameter;
daal::algorithms::decision_forest::training::interface1::Parameter::engine
engines::EnginePtr engine
Definition: decision_forest_training_parameter.h:146
daal::algorithms::decision_forest::training::interface1::Parameter::bootstrap
bool bootstrap
Definition: decision_forest_training_parameter.h:152
daal
Definition: algorithm_base_common.h:57
daal::algorithms::decision_forest::training::interface1::Parameter::maxTreeDepth
size_t maxTreeDepth
Definition: decision_forest_training_parameter.h:142
daal::algorithms::decision_forest::training::interface1::Parameter::impurityThreshold
double impurityThreshold
Definition: decision_forest_training_parameter.h:147
daal::algorithms::decision_forest::training::interface1::Parameter::minObservationsInLeafNode
size_t minObservationsInLeafNode
Definition: decision_forest_training_parameter.h:143
daal::algorithms::decision_forest::training::interface1::Parameter::resultsToCompute
DAAL_UINT64 resultsToCompute
Definition: decision_forest_training_parameter.h:150
daal::algorithms::decision_forest::training::VariableImportanceMode
VariableImportanceMode
Variable importance computation mode.
Definition: decision_forest_training_parameter.h:84
daal::algorithms::decision_forest::training::interface1::Parameter
Parameters for the decision forest algorithm.
Definition: decision_forest_training_parameter.h:121
daal::algorithms::decision_forest::training::interface1::Parameter::observationsPerTreeFraction
double observationsPerTreeFraction
Definition: decision_forest_training_parameter.h:137
daal::algorithms::decision_forest::training::interface1::Parameter::memorySavingMode
bool memorySavingMode
Definition: decision_forest_training_parameter.h:151
daal::algorithms::decision_forest::training::interface1::Parameter::nTrees
size_t nTrees
Definition: decision_forest_training_parameter.h:136
daal::algorithms::decision_forest::training::interface1::Parameter::featuresPerNode
size_t featuresPerNode
Definition: decision_forest_training_parameter.h:139
daal::algorithms::decision_forest::training::interface1::Parameter::varImportance
VariableImportanceMode varImportance
Definition: decision_forest_training_parameter.h:149
daal::algorithms::decision_forest::training::interface1::Parameter::seed
size_t seed
Definition: decision_forest_training_parameter.h:145
daal::algorithms::decision_forest::training::ResultToComputeId
ResultToComputeId
Definition: decision_forest_training_parameter.h:103