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,
106 computeOutOfBagErrorPerObservation = 0x00000002ULL
122 class DAAL_EXPORT Parameter
125 Parameter() : nTrees(100),
126 observationsPerTreeFraction(1.),
127 featuresPerNode(0), maxTreeDepth(0),
128 minObservationsInLeafNode(0),
130 impurityThreshold(0.),
133 memorySavingMode(
false),
135 engine(engines::mt19937::Batch<>::create()) {}
138 double observationsPerTreeFraction;
140 size_t featuresPerNode;
144 size_t minObservationsInLeafNode;
147 engines::EnginePtr engine;
148 double impurityThreshold;
150 VariableImportanceMode varImportance;
151 DAAL_UINT64 resultsToCompute;
152 bool memorySavingMode;
157 using interface1::Parameter;
daal::algorithms::decision_forest::training::interface1::Parameter::engine
engines::EnginePtr engine
Definition: decision_forest_training_parameter.h:147
daal::algorithms::decision_forest::training::interface1::Parameter::bootstrap
bool bootstrap
Definition: decision_forest_training_parameter.h:153
daal
Definition: algorithm_base_common.h:57
daal::algorithms::decision_forest::training::interface1::Parameter::maxTreeDepth
size_t maxTreeDepth
Definition: decision_forest_training_parameter.h:143
daal::algorithms::decision_forest::training::interface1::Parameter::impurityThreshold
double impurityThreshold
Definition: decision_forest_training_parameter.h:148
daal::algorithms::decision_forest::training::interface1::Parameter::minObservationsInLeafNode
size_t minObservationsInLeafNode
Definition: decision_forest_training_parameter.h:144
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:122
daal::algorithms::decision_forest::training::interface1::Parameter::observationsPerTreeFraction
double observationsPerTreeFraction
Definition: decision_forest_training_parameter.h:138
daal::algorithms::decision_forest::training::interface1::Parameter::memorySavingMode
bool memorySavingMode
Definition: decision_forest_training_parameter.h:152
daal::algorithms::engines::mt19937::interface1::Batch
Provides methods for mt19937 engine computations in the batch processing mode.
Definition: mt19937.h:113
daal::algorithms::decision_forest::training::interface1::Parameter::nTrees
size_t nTrees
Definition: decision_forest_training_parameter.h:137
daal::algorithms::decision_forest::training::interface1::Parameter::featuresPerNode
size_t featuresPerNode
Definition: decision_forest_training_parameter.h:140
daal::algorithms::decision_forest::training::interface1::Parameter::varImportance
VariableImportanceMode varImportance
Definition: decision_forest_training_parameter.h:150
daal::algorithms::decision_forest::training::interface1::Parameter::seed
size_t seed
Definition: decision_forest_training_parameter.h:146
daal::algorithms::decision_forest::training::ResultToComputeId
ResultToComputeId
Definition: decision_forest_training_parameter.h:103