24 #ifndef __GBT_REGRESSION_MODEL_H__
25 #define __GBT_REGRESSION_MODEL_H__
27 #include "algorithms/regression/regression_model.h"
28 #include "algorithms/regression/tree_traverse.h"
29 #include "algorithms/tree_utils/tree_utils_regression.h"
67 class DAAL_EXPORT Model :
public algorithms::regression::Model
70 DECLARE_MODEL(Model, algorithms::regression::Model);
77 static services::SharedPtr<Model> create(
size_t nFeatures, services::Status *stat = NULL);
84 virtual size_t numberOfTrees()
const = 0;
92 virtual void traverseDF(
size_t iTree, algorithms::regression::TreeNodeVisitor& visitor)
const = 0;
100 virtual void traverseBF(
size_t iTree, algorithms::regression::TreeNodeVisitor& visitor)
const = 0;
105 virtual void clear() = 0;
112 virtual void traverseDFS(
size_t iTree, tree_utils::regression::TreeNodeVisitor& visitor)
const = 0;
119 virtual void traverseBFS(
size_t iTree, tree_utils::regression::TreeNodeVisitor& visitor)
const = 0;
125 virtual size_t getNumberOfTrees()
const = 0;
130 typedef services::SharedPtr<Model> ModelPtr;
131 typedef services::SharedPtr<const Model> ModelConstPtr;
135 using interface1::Model;
136 using interface1::ModelPtr;
137 using interface1::ModelConstPtr;
daal::algorithms::gbt::regression::interface1::Model
Base class for models trained with the gradient boosted trees regression algorithm ...
Definition: gbt_regression_model.h:67