22 #ifndef __DECISION_FOREST_TRAINING_PARAMETER_H__
23 #define __DECISION_FOREST_TRAINING_PARAMETER_H__
25 #include "algorithms/algorithm.h"
26 #include "data_management/data/numeric_table.h"
27 #include "data_management/data/data_serialize.h"
28 #include "services/daal_defines.h"
29 #include "algorithms/engines/mt2203/mt2203.h"
43 namespace decision_forest
58 enum VariableImportanceMode
77 enum ResultToComputeId
79 computeOutOfBagError = 0x00000001ULL,
80 computeOutOfBagErrorPerObservation = 0x00000002ULL
96 class DAAL_EXPORT Parameter
99 Parameter() : nTrees(100),
100 observationsPerTreeFraction(1.),
101 featuresPerNode(0), maxTreeDepth(0),
102 minObservationsInLeafNode(0),
104 impurityThreshold(0.),
107 memorySavingMode(
false),
109 engine(engines::mt2203::Batch<>::create()) {}
112 double observationsPerTreeFraction;
114 size_t featuresPerNode;
118 size_t minObservationsInLeafNode;
121 engines::EnginePtr engine;
122 double impurityThreshold;
124 VariableImportanceMode varImportance;
125 DAAL_UINT64 resultsToCompute;
126 bool memorySavingMode;
131 using interface1::Parameter;
daal::algorithms::decision_forest::training::interface1::Parameter::engine
engines::EnginePtr engine
Definition: decision_forest_training_parameter.h:121
daal::algorithms::decision_forest::training::interface1::Parameter::bootstrap
bool bootstrap
Definition: decision_forest_training_parameter.h:127
daal
Definition: algorithm_base_common.h:31
daal::algorithms::decision_forest::training::interface1::Parameter::maxTreeDepth
size_t maxTreeDepth
Definition: decision_forest_training_parameter.h:117
daal::algorithms::decision_forest::training::interface1::Parameter::impurityThreshold
double impurityThreshold
Definition: decision_forest_training_parameter.h:122
daal::algorithms::decision_forest::training::interface1::Parameter::minObservationsInLeafNode
size_t minObservationsInLeafNode
Definition: decision_forest_training_parameter.h:118
daal::algorithms::decision_forest::training::interface1::Parameter::resultsToCompute
DAAL_UINT64 resultsToCompute
Definition: decision_forest_training_parameter.h:125
daal::algorithms::decision_forest::training::VariableImportanceMode
VariableImportanceMode
Variable importance computation mode.
Definition: decision_forest_training_parameter.h:58
daal::algorithms::decision_forest::training::interface1::Parameter
Parameters for the decision forest algorithm.
Definition: decision_forest_training_parameter.h:96
daal::algorithms::decision_forest::training::interface1::Parameter::observationsPerTreeFraction
double observationsPerTreeFraction
Definition: decision_forest_training_parameter.h:112
daal::algorithms::decision_forest::training::interface1::Parameter::memorySavingMode
bool memorySavingMode
Definition: decision_forest_training_parameter.h:126
daal::algorithms::decision_forest::training::interface1::Parameter::nTrees
size_t nTrees
Definition: decision_forest_training_parameter.h:111
daal::algorithms::decision_forest::training::interface1::Parameter::featuresPerNode
size_t featuresPerNode
Definition: decision_forest_training_parameter.h:114
daal::algorithms::decision_forest::training::interface1::Parameter::varImportance
VariableImportanceMode varImportance
Definition: decision_forest_training_parameter.h:124
daal::algorithms::decision_forest::training::interface1::Parameter::seed
size_t seed
Definition: decision_forest_training_parameter.h:120
daal::algorithms::decision_forest::training::ResultToComputeId
ResultToComputeId
Definition: decision_forest_training_parameter.h:77