24 #ifndef __STUMP_MODEL_H__
25 #define __STUMP_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/weak_learner/weak_learner_model.h"
58 class DAAL_EXPORT Model :
public weak_learner::Model
61 DECLARE_MODEL(Model, classifier::Model);
70 template<
typename modelFPType>
71 DAAL_EXPORT Model(
size_t nFeatures, modelFPType dummy);
80 template<
typename modelFPType>
81 DAAL_EXPORT
static services::SharedPtr<Model> create(
size_t nFeatures, services::Status *stat = NULL);
92 size_t getSplitFeature();
98 void setSplitFeature(
size_t splitFeature);
104 template<
typename modelFPType>
105 DAAL_EXPORT modelFPType getSplitValue();
111 template<
typename modelFPType>
112 DAAL_EXPORT
void setSplitValue(modelFPType splitValue);
118 template<
typename modelFPType>
119 DAAL_EXPORT modelFPType getLeftSubsetAverage();
125 template<
typename modelFPType>
126 DAAL_EXPORT
void setLeftSubsetAverage(modelFPType leftSubsetAverage);
132 template<
typename modelFPType>
133 DAAL_EXPORT modelFPType getRightSubsetAverage();
139 template<
typename modelFPType>
140 DAAL_EXPORT
void setRightSubsetAverage(modelFPType rightSubsetAverage);
146 size_t getNumberOfFeatures() const DAAL_C11_OVERRIDE {
return _nFeatures; }
151 size_t _splitFeature;
152 services::SharedPtr<data_management::Matrix<double> > _values;
157 template<
typename modelFPType>
158 DAAL_EXPORT Model(
size_t nFeatures, modelFPType dummy, services::Status &st);
161 template<
typename Archive,
bool onDeserialize>
162 services::Status serialImpl(Archive *arch)
164 services::Status st = classifier::Model::serialImpl<Archive, onDeserialize>(arch);
167 arch->set(_nFeatures);
168 arch->set(_splitFeature);
169 arch->setSharedPtrObj(_values);
174 typedef services::SharedPtr<Model> ModelPtr;
176 using interface1::Model;
177 using interface1::ModelPtr;
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:54
daal::algorithms::stump::interface1::Model::_nFeatures
size_t _nFeatures
Definition: stump_model.h:149
daal::algorithms::stump::interface1::Model::_splitFeature
size_t _splitFeature
Definition: stump_model.h:151
daal::algorithms::stump::interface1::Model
Model of the classifier trained by the stump::training::Batch algorithm.
Definition: stump_model.h:58
daal::algorithms::stump::interface1::Model::_values
services::SharedPtr< data_management::Matrix< double > > _values
Definition: stump_model.h:152
daal::algorithms::stump::interface1::Model::getNumberOfFeatures
size_t getNumberOfFeatures() const DAAL_C11_OVERRIDE
Definition: stump_model.h:146