22 #ifndef __STUMP_MODEL_H__
23 #define __STUMP_MODEL_H__
25 #include "algorithms/algorithm.h"
26 #include "data_management/data/homogen_numeric_table.h"
27 #include "data_management/data/matrix.h"
28 #include "algorithms/weak_learner/weak_learner_model.h"
56 class DAAL_EXPORT Model :
public weak_learner::Model
59 DECLARE_MODEL(Model, classifier::Model);
68 template<
typename modelFPType>
69 DAAL_EXPORT Model(
size_t nFeatures, modelFPType dummy);
78 template<
typename modelFPType>
79 DAAL_EXPORT
static services::SharedPtr<Model> create(
size_t nFeatures, services::Status *stat = NULL);
90 size_t getSplitFeature();
96 void setSplitFeature(
size_t splitFeature);
102 template<
typename modelFPType>
103 DAAL_EXPORT modelFPType getSplitValue();
109 template<
typename modelFPType>
110 DAAL_EXPORT
void setSplitValue(modelFPType splitValue);
116 template<
typename modelFPType>
117 DAAL_EXPORT modelFPType getLeftSubsetAverage();
123 template<
typename modelFPType>
124 DAAL_EXPORT
void setLeftSubsetAverage(modelFPType leftSubsetAverage);
130 template<
typename modelFPType>
131 DAAL_EXPORT modelFPType getRightSubsetAverage();
137 template<
typename modelFPType>
138 DAAL_EXPORT
void setRightSubsetAverage(modelFPType rightSubsetAverage);
144 size_t getNumberOfFeatures() const DAAL_C11_OVERRIDE {
return _nFeatures; }
149 size_t _splitFeature;
150 services::SharedPtr<data_management::Matrix<double> > _values;
155 template<
typename modelFPType>
156 DAAL_EXPORT Model(
size_t nFeatures, modelFPType dummy, services::Status &st);
159 template<
typename Archive,
bool onDeserialize>
160 services::Status serialImpl(Archive *arch)
162 services::Status st = classifier::Model::serialImpl<Archive, onDeserialize>(arch);
165 arch->set(_nFeatures);
166 arch->set(_splitFeature);
167 arch->setSharedPtrObj(_values);
172 typedef services::SharedPtr<Model> ModelPtr;
174 using interface1::Model;
175 using interface1::ModelPtr;
daal
Definition: algorithm_base_common.h:31
daal::algorithms::stump::interface1::Model
Model of the classifier trained by the stump::training::Batch algorithm.
Definition: stump_model.h:56
daal::algorithms::stump::interface1::Model::_splitFeature
size_t _splitFeature
Definition: stump_model.h:149
daal::algorithms::stump::interface1::Model::getNumberOfFeatures
size_t getNumberOfFeatures() const DAAL_C11_OVERRIDE
Definition: stump_model.h:144
daal::algorithms::stump::interface1::Model::_nFeatures
size_t _nFeatures
Definition: stump_model.h:147
daal::algorithms::stump::interface1::Model::_values
services::SharedPtr< data_management::Matrix< double > > _values
Definition: stump_model.h:150
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:52