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::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::weak_learner::interface1::Model
Base class for the weak learner model
Definition: weak_learner_model.h:90
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::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::classifier::interface1::Model
Base class for the model of the classification algorithm.
Definition: classifier_model.h:95
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