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

gbt_regression_model.h
1 /* file: gbt_regression_model.h */
2 /*******************************************************************************
3 * Copyright 2014-2018 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 
81  virtual size_t numberOfTrees() const = 0;
82 
89  virtual void traverseDF(size_t iTree, algorithms::regression::TreeNodeVisitor& visitor) const = 0;
90 
97  virtual void traverseBF(size_t iTree, algorithms::regression::TreeNodeVisitor& visitor) const = 0;
98 
102  virtual void clear() = 0;
103 
109  virtual void traverseDFS(size_t iTree, tree_utils::regression::TreeNodeVisitor& visitor) const = 0;
110 
116  virtual void traverseBFS(size_t iTree, tree_utils::regression::TreeNodeVisitor& visitor) const = 0;
117 
118 protected:
119  Model();
120 };
121 typedef services::SharedPtr<Model> ModelPtr;
122 typedef services::SharedPtr<const Model> ModelConstPtr;
123 
125 } // namespace interface1
126 using interface1::Model;
127 using interface1::ModelPtr;
128 using interface1::ModelConstPtr;
129 
130 }
131 }
132 }
133 }
134 #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.