24 #ifndef __DECISION_FOREST_TRAINING_PARAMETER_H__
25 #define __DECISION_FOREST_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 "services/daal_defines.h"
31 #include "algorithms/engines/mt2203/mt2203.h"
45 namespace decision_forest
60 enum VariableImportanceMode
79 enum ResultToComputeId
81 computeOutOfBagError = 0x00000001ULL,
82 computeOutOfBagErrorPerObservation = 0x00000002ULL
98 class DAAL_EXPORT Parameter
101 Parameter() : nTrees(100),
102 observationsPerTreeFraction(1.),
103 featuresPerNode(0), maxTreeDepth(0),
104 minObservationsInLeafNode(0),
106 impurityThreshold(0.),
109 memorySavingMode(
false),
111 engine(engines::mt2203::Batch<>::create()) {}
114 double observationsPerTreeFraction;
116 size_t featuresPerNode;
120 size_t minObservationsInLeafNode;
123 engines::EnginePtr engine;
124 double impurityThreshold;
126 VariableImportanceMode varImportance;
127 DAAL_UINT64 resultsToCompute;
128 bool memorySavingMode;
133 using interface1::Parameter;
daal::algorithms::decision_forest::training::interface1::Parameter::maxTreeDepth
size_t maxTreeDepth
Definition: decision_forest_training_parameter.h:119
daal::algorithms::decision_forest::training::VariableImportanceMode
VariableImportanceMode
Variable importance computation mode.
Definition: decision_forest_training_parameter.h:60
daal::algorithms::decision_forest::training::ResultToComputeId
ResultToComputeId
Definition: decision_forest_training_parameter.h:79
daal::algorithms::decision_forest::training::interface1::Parameter::nTrees
size_t nTrees
Definition: decision_forest_training_parameter.h:113
daal::algorithms::decision_forest::training::interface1::Parameter
Parameters for the decision forest algorithm.
Definition: decision_forest_training_parameter.h:98
daal::algorithms::decision_forest::training::interface1::Parameter::observationsPerTreeFraction
double observationsPerTreeFraction
Definition: decision_forest_training_parameter.h:114
daal::algorithms::decision_forest::training::interface1::Parameter::memorySavingMode
bool memorySavingMode
Definition: decision_forest_training_parameter.h:128
daal::algorithms::decision_forest::training::interface1::Parameter::seed
size_t seed
Definition: decision_forest_training_parameter.h:122
daal::algorithms::decision_forest::training::interface1::Parameter::minObservationsInLeafNode
size_t minObservationsInLeafNode
Definition: decision_forest_training_parameter.h:120
daal::algorithms::decision_forest::training::interface1::Parameter::varImportance
VariableImportanceMode varImportance
Definition: decision_forest_training_parameter.h:126
daal::algorithms::decision_forest::training::interface1::Parameter::bootstrap
bool bootstrap
Definition: decision_forest_training_parameter.h:129
daal::algorithms::decision_forest::training::interface1::Parameter::resultsToCompute
DAAL_UINT64 resultsToCompute
Definition: decision_forest_training_parameter.h:127
daal::algorithms::decision_forest::training::interface1::Parameter::engine
engines::EnginePtr engine
Definition: decision_forest_training_parameter.h:123
daal::algorithms::decision_forest::training::interface1::Parameter::impurityThreshold
double impurityThreshold
Definition: decision_forest_training_parameter.h:124
daal::algorithms::decision_forest::training::interface1::Parameter::featuresPerNode
size_t featuresPerNode
Definition: decision_forest_training_parameter.h:116