C++ API Reference for Intel® Data Analytics Acceleration Library 2018 Update 3

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 
28 namespace daal
29 {
30 namespace algorithms
31 {
32 namespace gbt
33 {
42 namespace regression
43 {
47 namespace interface1
48 {
49 
64 class DAAL_EXPORT Model : public algorithms::regression::Model
65 {
66 public:
67  DECLARE_MODEL(Model, algorithms::regression::Model);
68 
74  static services::SharedPtr<Model> create(size_t nFeatures, services::Status *stat = NULL);
75 
80  virtual size_t numberOfTrees() const = 0;
81 
87  virtual void traverseDF(size_t iTree, algorithms::regression::TreeNodeVisitor& visitor) const = 0;
88 
94  virtual void traverseBF(size_t iTree, algorithms::regression::TreeNodeVisitor& visitor) const = 0;
95 
99  virtual void clear() = 0;
100 
101 protected:
102  Model();
103 };
104 typedef services::SharedPtr<Model> ModelPtr;
105 typedef services::SharedPtr<const Model> ModelConstPtr;
106 
108 } // namespace interface1
109 using interface1::Model;
110 using interface1::ModelPtr;
111 using interface1::ModelConstPtr;
112 
113 }
114 }
115 }
116 }
117 #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:64

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