24 #ifndef __GBT_CLASSIFICATION_MODEL_H__
25 #define __GBT_CLASSIFICATION_MODEL_H__
27 #include "algorithms/classifier/classifier_model.h"
28 #include "algorithms/regression/tree_traverse.h"
29 #include "algorithms/tree_utils/tree_utils_regression.h"
45 namespace classification
64 class DAAL_EXPORT Model :
public classifier::Model
67 DECLARE_MODEL(Model, classifier::Model)
74 static services::SharedPtr<Model> create(
size_t nFeatures, services::Status *stat = NULL);
81 virtual size_t numberOfTrees()
const = 0;
90 virtual void traverseDF(
size_t iTree, daal::algorithms::regression::TreeNodeVisitor& visitor)
const = 0;
99 virtual void traverseBF(
size_t iTree, daal::algorithms::regression::TreeNodeVisitor& visitor)
const = 0;
104 virtual void clear() = 0;
111 virtual void traverseDFS(
size_t iTree, tree_utils::regression::TreeNodeVisitor& visitor)
const = 0;
118 virtual void traverseBFS(
size_t iTree, tree_utils::regression::TreeNodeVisitor& visitor)
const = 0;
124 virtual size_t getNumberOfTrees()
const = 0;
127 Model() : classifier::Model()
131 typedef services::SharedPtr<Model> ModelPtr;
133 using interface1::Model;
134 using interface1::ModelPtr;
daal::algorithms::gbt::classification::interface1::Model
Model of the classifier trained by the gbt::training::Batch algorithm.
Definition: gbt_classification_model.h:64