24 #ifndef __STUMP_REGRESSION_MODEL_H__
25 #define __STUMP_REGRESSION_MODEL_H__
27 #include "algorithms/algorithm.h"
28 #include "data_management/data/homogen_numeric_table.h"
29 #include "data_management/data/matrix.h"
30 #include "algorithms/regression/regression_model.h"
31 #include "algorithms/decision_tree/decision_tree_regression_model.h"
51 enum VariableImportanceMode
73 struct DAAL_EXPORT Parameter :
public daal::algorithms::Parameter
78 Parameter() : daal::algorithms::Parameter(), varImportance(none) {}
83 services::Status check() const DAAL_C11_OVERRIDE;
85 VariableImportanceMode varImportance;
98 class DAAL_EXPORT Model : public daal::algorithms::decision_tree::regression::Model
101 DECLARE_MODEL_IFACE(Model, decision_tree::regression::Model);
109 static services::SharedPtr<Model> create(services::Status *stat = NULL);
122 size_t getSplitFeature()
const;
128 template<
typename modelFPType>
129 DAAL_EXPORT modelFPType getSplitValue();
135 template<
typename modelFPType>
136 DAAL_EXPORT modelFPType getLeftValue();
142 template<
typename modelFPType>
143 DAAL_EXPORT modelFPType getRightValue();
146 Model(services::Status &st);
148 services::Status serializeImpl(data_management::InputDataArchive * arch) DAAL_C11_OVERRIDE;
150 services::Status deserializeImpl(
const data_management::OutputDataArchive * arch) DAAL_C11_OVERRIDE;
154 typedef services::SharedPtr<Model> ModelPtr;
155 typedef services::SharedPtr<const Model> ModelConstPtr;
159 using interface1::Parameter;
160 using interface1::Model;
161 using interface1::ModelPtr;
162 using interface1::ModelConstPtr;
daal::algorithms::stump::regression::interface1::Model::Model
Model()
Definition: stump_regression_model.h:114
daal::algorithms::decision_forest::training::VariableImportanceMode
VariableImportanceMode
Variable importance computation mode.
Definition: decision_forest_training_parameter.h:60
daal::algorithms::stump::regression::interface1::Parameter::Parameter
Parameter()
Definition: stump_regression_model.h:78
daal::algorithms::interface1::Parameter
Base class to represent computation parameters. Algorithm-specific parameters are represented as deri...
Definition: algorithm_types.h:62
daal::algorithms::stump::regression::interface1::Parameter
Stump algorithm parameters.
Definition: stump_regression_model.h:73
daal::algorithms::stump::interface1::Model
Model of the classifier trained by the stump::training::Batch algorithm.
Definition: stump_model.h:58
daal::algorithms::stump::regression::interface1::Model
Model of the regression trained by the stump::regression::training::Batch algorithm.
Definition: stump_regression_model.h:98