22 #ifndef __DECISION_TREE_REGRESSION_MODEL_H__
23 #define __DECISION_TREE_REGRESSION_MODEL_H__
25 #include "data_management/data/numeric_table.h"
26 #include "data_management/data/aos_numeric_table.h"
27 #include "data_management/data/soa_numeric_table.h"
28 #include "data_management/data/homogen_numeric_table.h"
29 #include "algorithms/model.h"
30 #include "decision_tree_model.h"
31 #include "algorithms/regression/regression_model.h"
32 #include "algorithms/regression/tree_traverse.h"
33 #include "algorithms/tree_utils/tree_utils_regression.h"
42 namespace decision_tree
71 struct DAAL_EXPORT Parameter :
public daal::algorithms::Parameter
76 Parameter() : daal::algorithms::Parameter(),
77 pruning(reducedErrorPruning), maxTreeDepth(0), minObservationsInLeafNodes(5)
83 services::Status check() const DAAL_C11_OVERRIDE;
87 size_t minObservationsInLeafNodes;
100 class DAAL_EXPORT Model : public algorithms::regression::Model
103 DECLARE_MODEL_IFACE(Model, algorithms::regression::Model);
114 typedef services::SharedPtr<ModelImpl> ModelImplPtr;
120 const ModelImpl * impl()
const {
return _impl.get(); }
126 ModelImpl * impl() {
return _impl.get(); }
132 static services::SharedPtr<Model> create(services::Status *stat = NULL);
137 virtual size_t getNumberOfFeatures() const DAAL_C11_OVERRIDE;
144 void traverseDF(algorithms::regression::TreeNodeVisitor& visitor) const;
151 void traverseBF(algorithms::regression::TreeNodeVisitor& visitor) const;
157 void traverseDFS(tree_utils::regression::TreeNodeVisitor& visitor) const;
163 void traverseBFS(tree_utils::regression::TreeNodeVisitor& visitor) const;
166 Model(services::Status &st);
168 services::Status serializeImpl(data_management::InputDataArchive * arch) DAAL_C11_OVERRIDE;
170 services::Status deserializeImpl(const data_management::OutputDataArchive * arch) DAAL_C11_OVERRIDE;
176 typedef services::SharedPtr<Model> ModelPtr;
177 typedef services::SharedPtr<const Model> ModelConstPtr;
181 using interface1::Parameter;
182 using interface1::Model;
183 using interface1::ModelPtr;
184 using interface1::ModelConstPtr;
daal::algorithms::decision_tree::regression::interface1::Parameter::Parameter
Parameter()
Definition: decision_tree_regression_model.h:76
daal
Definition: algorithm_base_common.h:31
daal::algorithms::decision_tree::reducedErrorPruning
Definition: decision_tree_model.h:50
daal::algorithms::decision_tree::regression::interface1::Parameter
Decision tree algorithm parameters.
Definition: decision_tree_regression_model.h:71
daal::algorithms::decision_tree::regression::interface1::Model::impl
const ModelImpl * impl() const
Definition: decision_tree_regression_model.h:120
daal::algorithms::decision_tree::Pruning
Pruning
Pruning method for Decision tree algorithm.
Definition: decision_tree_model.h:47
daal::algorithms::decision_tree::regression::interface1::Model
Base class for models trained with the Decision tree algorithm
Definition: decision_tree_regression_model.h:100
daal::algorithms::decision_tree::regression::interface1::Model::impl
ModelImpl * impl()
Definition: decision_tree_regression_model.h:126