48 #ifndef __STUMP_MODEL_H__
49 #define __STUMP_MODEL_H__
51 #include "algorithms/algorithm.h"
52 #include "data_management/data/homogen_numeric_table.h"
53 #include "data_management/data/matrix.h"
54 #include "algorithms/weak_learner/weak_learner_model.h"
82 class DAAL_EXPORT Model :
public weak_learner::Model
85 DECLARE_MODEL(Model, classifier::Model);
94 template<
typename modelFPType>
95 DAAL_EXPORT Model(
size_t nFeatures, modelFPType dummy);
104 template<
typename modelFPType>
105 DAAL_EXPORT
static services::SharedPtr<Model> create(
size_t nFeatures, services::Status *stat = NULL);
116 size_t getSplitFeature();
122 void setSplitFeature(
size_t splitFeature);
128 template<
typename modelFPType>
129 DAAL_EXPORT modelFPType getSplitValue();
135 template<
typename modelFPType>
136 DAAL_EXPORT
void setSplitValue(modelFPType splitValue);
142 template<
typename modelFPType>
143 DAAL_EXPORT modelFPType getLeftSubsetAverage();
149 template<
typename modelFPType>
150 DAAL_EXPORT
void setLeftSubsetAverage(modelFPType leftSubsetAverage);
156 template<
typename modelFPType>
157 DAAL_EXPORT modelFPType getRightSubsetAverage();
163 template<
typename modelFPType>
164 DAAL_EXPORT
void setRightSubsetAverage(modelFPType rightSubsetAverage);
170 size_t getNumberOfFeatures() const DAAL_C11_OVERRIDE {
return _nFeatures; }
175 size_t _splitFeature;
176 services::SharedPtr<data_management::Matrix<double> > _values;
181 template<
typename modelFPType>
182 DAAL_EXPORT Model(
size_t nFeatures, modelFPType dummy, services::Status &st);
185 template<
typename Archive,
bool onDeserialize>
186 services::Status serialImpl(Archive *arch)
188 services::Status st = classifier::Model::serialImpl<Archive, onDeserialize>(arch);
191 arch->set(_nFeatures);
192 arch->set(_splitFeature);
193 arch->setSharedPtrObj(_values);
198 typedef services::SharedPtr<Model> ModelPtr;
200 using interface1::Model;
201 using interface1::ModelPtr;
daal
Definition: algorithm_base_common.h:57
daal::algorithms::stump::interface1::Model
Model of the classifier trained by the stump::training::Batch algorithm.
Definition: stump_model.h:82
daal::algorithms::stump::interface1::Model::_splitFeature
size_t _splitFeature
Definition: stump_model.h:175
daal::algorithms::stump::interface1::Model::getNumberOfFeatures
size_t getNumberOfFeatures() const DAAL_C11_OVERRIDE
Definition: stump_model.h:170
daal::algorithms::stump::interface1::Model::_nFeatures
size_t _nFeatures
Definition: stump_model.h:173
daal::algorithms::stump::interface1::Model::_values
services::SharedPtr< data_management::Matrix< double > > _values
Definition: stump_model.h:176
daal::algorithms::interface1::Model
The base class for the classes that represent the models, such as linear_regression::Model or svm::Mo...
Definition: model.h:78