C++ API Reference for Intel® Data Analytics Acceleration Library 2019 Update 5

gbt_regression_model.h
1 /* file: gbt_regression_model.h */
2 /*******************************************************************************
3 * Copyright 2014-2019 Intel Corporation.
4 *
5 * This software and the related documents are Intel copyrighted materials, and
6 * your use of them is governed by the express license under which they were
7 * provided to you (License). Unless the License provides otherwise, you may not
8 * use, modify, copy, publish, distribute, disclose or transmit this software or
9 * the related documents without Intel's prior written permission.
10 *
11 * This software and the related documents are provided as is, with no express
12 * or implied warranties, other than those that are expressly stated in the
13 * License.
14 *******************************************************************************/
15 
16 /*
17 //++
18 // Implementation of the class defining the gradient boosted trees regression model
19 //--
20 */
21 
22 #ifndef __GBT_REGRESSION_MODEL_H__
23 #define __GBT_REGRESSION_MODEL_H__
24 
25 #include "algorithms/regression/regression_model.h"
26 #include "algorithms/regression/tree_traverse.h"
27 #include "algorithms/tree_utils/tree_utils_regression.h"
28 
29 namespace daal
30 {
31 namespace algorithms
32 {
33 namespace gbt
34 {
43 namespace regression
44 {
48 namespace interface1
49 {
50 
65 class DAAL_EXPORT Model : public algorithms::regression::Model
66 {
67 public:
68  DECLARE_MODEL(Model, algorithms::regression::Model);
69 
75  static services::SharedPtr<Model> create(size_t nFeatures, services::Status *stat = NULL);
76 
82  virtual size_t numberOfTrees() const = 0;
83 
90  virtual void traverseDF(size_t iTree, algorithms::regression::TreeNodeVisitor& visitor) const = 0;
91 
98  virtual void traverseBF(size_t iTree, algorithms::regression::TreeNodeVisitor& visitor) const = 0;
99 
103  virtual void clear() = 0;
104 
110  virtual void traverseDFS(size_t iTree, tree_utils::regression::TreeNodeVisitor& visitor) const = 0;
111 
117  virtual void traverseBFS(size_t iTree, tree_utils::regression::TreeNodeVisitor& visitor) const = 0;
118 
123  virtual size_t getNumberOfTrees() const = 0;
124 
125 protected:
126  Model();
127 };
128 typedef services::SharedPtr<Model> ModelPtr;
129 typedef services::SharedPtr<const Model> ModelConstPtr;
130 
132 } // namespace interface1
133 using interface1::Model;
134 using interface1::ModelPtr;
135 using interface1::ModelConstPtr;
136 
137 }
138 }
139 }
140 }
141 #endif
daal
Definition: algorithm_base_common.h:31
daal::algorithms::gbt::regression::interface1::Model
Base class for models trained with the gradient boosted trees regression algorithm ...
Definition: gbt_regression_model.h:65

For more complete information about compiler optimizations, see our Optimization Notice.