48 #ifndef __GBT_CLASSIFICATION_MODEL_H__
49 #define __GBT_CLASSIFICATION_MODEL_H__
51 #include "algorithms/classifier/classifier_model.h"
52 #include "algorithms/regression/tree_traverse.h"
54 using namespace daal::services;
70 namespace classification
89 class DAAL_EXPORT Model :
public classifier::Model
92 DECLARE_MODEL(Model, classifier::Model)
99 static services::SharedPtr<Model> create(
size_t nFeatures, services::Status *stat = NULL);
105 virtual size_t numberOfTrees()
const = 0;
112 virtual void traverseDF(
size_t iTree, regression::TreeNodeVisitor& visitor)
const = 0;
119 virtual void traverseBF(
size_t iTree, regression::TreeNodeVisitor& visitor)
const = 0;
122 Model() : classifier::Model()
126 typedef services::SharedPtr<Model> ModelPtr;
128 using interface1::Model;
129 using interface1::ModelPtr;
daal
Definition: algorithm_base_common.h:57
daal::algorithms::gbt::classification::interface1::Model
Model of the classifier trained by the gbt::training::Batch algorithm.
Definition: gbt_classification_model.h:89
daal::services
Contains classes that implement service functionality, including error handling, memory allocation...
Definition: collection.h:51