48 #ifndef __DECISION_TREE_REGRESSION_MODEL_H__ 49 #define __DECISION_TREE_REGRESSION_MODEL_H__ 51 #include "data_management/data/numeric_table.h" 52 #include "data_management/data/aos_numeric_table.h" 53 #include "data_management/data/soa_numeric_table.h" 54 #include "data_management/data/homogen_numeric_table.h" 55 #include "algorithms/model.h" 56 #include "decision_tree_model.h" 57 #include "algorithms/regression/regression_model.h" 58 #include "algorithms/regression/tree_traverse.h" 67 namespace decision_tree
96 struct DAAL_EXPORT Parameter :
public daal::algorithms::Parameter
101 Parameter() : daal::algorithms::Parameter(),
102 pruning(reducedErrorPruning), maxTreeDepth(0), minObservationsInLeafNodes(5)
108 services::Status check()
const DAAL_C11_OVERRIDE;
112 size_t minObservationsInLeafNodes;
125 class DAAL_EXPORT Model :
public algorithms::regression::Model
128 DECLARE_MODEL_IFACE(Model, algorithms::regression::Model);
139 typedef services::SharedPtr<ModelImpl> ModelImplPtr;
145 const ModelImpl * impl()
const {
return _impl.get(); }
151 ModelImpl * impl() {
return _impl.get(); }
157 static services::SharedPtr<Model> create(services::Status *stat = NULL);
162 virtual size_t getNumberOfFeatures()
const DAAL_C11_OVERRIDE;
168 void traverseDF(algorithms::regression::TreeNodeVisitor& visitor)
const;
174 void traverseBF(algorithms::regression::TreeNodeVisitor& visitor)
const;
177 Model(services::Status &st);
179 services::Status serializeImpl(data_management::InputDataArchive * arch) DAAL_C11_OVERRIDE;
181 services::Status deserializeImpl(
const data_management::OutputDataArchive * arch) DAAL_C11_OVERRIDE;
187 typedef services::SharedPtr<Model> ModelPtr;
188 typedef services::SharedPtr<const Model> ModelConstPtr;
192 using interface1::Parameter;
193 using interface1::Model;
194 using interface1::ModelPtr;
195 using interface1::ModelConstPtr;
daal::algorithms::decision_tree::regression::interface1::Model::impl
const ModelImpl * impl() const
Definition: decision_tree_regression_model.h:145
daal::algorithms::decision_tree::regression::interface1::Parameter::Parameter
Parameter()
Definition: decision_tree_regression_model.h:101
daal::services::interface1::Status
Class that holds the results of API calls. In case of API routine failure it contains the list of err...
Definition: error_handling.h:491
daal
Definition: algorithm_base_common.h:57
daal::algorithms::regression::interface1::TreeNodeVisitor
Interface of abstract visitor used in tree traversal methods.
Definition: regression/tree_traverse.h:86
daal::algorithms::decision_tree::regression::interface1::Parameter::minObservationsInLeafNodes
size_t minObservationsInLeafNodes
Definition: decision_tree_regression_model.h:112
daal::algorithms::decision_tree::regression::interface1::Parameter
Decision tree algorithm parameters.
Definition: decision_tree_regression_model.h:96
daal::algorithms::interface1::Parameter
Base class to represent computation parameters. Algorithm-specific parameters are represented as deri...
Definition: algorithm_types.h:86
daal::services::interface1::SharedPtr
Shared pointer that retains shared ownership of an object through a pointer. Several SharedPtr object...
Definition: daal_shared_ptr.h:187
daal::algorithms::decision_tree::reducedErrorPruning
Definition: decision_tree_model.h:76
daal::algorithms::regression::interface1::Model
Base class for models trained with the regression algorithm
Definition: regression_model.h:89
daal::data_management::interface1::OutputDataArchive
Provides methods to restore an object from its serialized counterpart and access the restored object...
Definition: data_archive.h:978
daal::algorithms::decision_tree::regression::interface1::Model
Base class for models trained with the Decision tree algorithm
Definition: decision_tree_regression_model.h:125
daal::algorithms::decision_tree::regression::interface1::Model::impl
ModelImpl * impl()
Definition: decision_tree_regression_model.h:151
daal::algorithms::decision_tree::regression::interface1::Parameter::pruning
Pruning pruning
Definition: decision_tree_regression_model.h:110
daal::algorithms::decision_tree::Pruning
Pruning
Pruning method for Decision tree algorithm.
Definition: decision_tree_model.h:73
daal::algorithms::decision_tree::regression::interface1::Parameter::maxTreeDepth
size_t maxTreeDepth
Definition: decision_tree_regression_model.h:111