22 #ifndef __BOOSTING_MODEL_H__
23 #define __BOOSTING_MODEL_H__
25 #include "algorithms/weak_learner/weak_learner_model.h"
26 #include "algorithms/weak_learner/weak_learner_training_batch.h"
27 #include "algorithms/weak_learner/weak_learner_predict.h"
28 #include "algorithms/stump/stump_training_batch.h"
29 #include "algorithms/stump/stump_predict.h"
30 #include "algorithms/classifier/classifier_model.h"
58 struct DAAL_EXPORT Parameter :
public classifier::Parameter
68 Parameter(
const services::SharedPtr<weak_learner::training::Batch>& wlTrainForParameter,
69 const services::SharedPtr<weak_learner::prediction::Batch>& wlPredictForParameter);
72 Parameter(
const Parameter& other) :weakLearnerTraining(other.weakLearnerTraining),
73 weakLearnerPrediction(other.weakLearnerPrediction){}
76 services::SharedPtr<weak_learner::training::Batch> weakLearnerTraining;
79 services::SharedPtr<weak_learner::prediction::Batch> weakLearnerPrediction;
81 services::Status check() const DAAL_C11_OVERRIDE;
95 class DAAL_EXPORT Model : public classifier::Model
103 Model(
size_t nFeatures = 0) : _models(new data_management::DataCollection()), _nFeatures(nFeatures) {}
111 size_t getNumberOfWeakLearners()
const;
118 weak_learner::ModelPtr getWeakLearnerModel(
size_t idx)
const;
124 void addWeakLearnerModel(weak_learner::ModelPtr model);
126 void clearWeakLearnerModels();
132 size_t getNumberOfFeatures() const DAAL_C11_OVERRIDE {
return _nFeatures; }
136 data_management::DataCollectionPtr _models;
138 template<
typename Archive,
bool onDeserialize>
139 services::Status serialImpl(Archive *arch)
141 classifier::Model::serialImpl<Archive, onDeserialize>(arch);
142 arch->set(_nFeatures);
143 arch->setSharedPtrObj(_models);
145 return services::Status();
148 Model(
size_t nFeatures, services::Status &st);
150 typedef services::SharedPtr<Model> ModelPtr;
153 using interface1::Parameter;
154 using interface1::Model;
155 using interface1::ModelPtr;
160 #endif // __BOOSTING_MODEL_H__
daal
Definition: algorithm_base_common.h:31
daal::algorithms::boosting::interface1::Parameter::weakLearnerPrediction
services::SharedPtr< weak_learner::prediction::Batch > weakLearnerPrediction
Definition: boosting_model.h:79
daal::algorithms::classifier::prediction::model
Definition: classifier_predict_types.h:66
daal::algorithms::boosting::interface1::Model::getNumberOfFeatures
size_t getNumberOfFeatures() const DAAL_C11_OVERRIDE
Definition: boosting_model.h:132
daal::algorithms::boosting::interface1::Parameter::Parameter
Parameter(const Parameter &other)
Definition: boosting_model.h:72
daal::algorithms::boosting::interface1::Model
Base class for boosting algorithm models. Contains a collection of weak learner models constructed du...
Definition: boosting_model.h:95
daal::algorithms::boosting::interface1::Parameter::weakLearnerTraining
services::SharedPtr< weak_learner::training::Batch > weakLearnerTraining
Definition: boosting_model.h:76
daal::algorithms::boosting::interface1::Parameter
Base class for parameters of the boosting algorithm
Definition: boosting_model.h:58
daal::algorithms::boosting::interface1::Model::Model
Model(size_t nFeatures=0)
Definition: boosting_model.h:103