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

gbt_classification_model.h
1 /* file: gbt_classification_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 class defining gradient boosted trees classification model.
19 //--
20 */
21 
22 #ifndef __GBT_CLASSIFICATION_MODEL_H__
23 #define __GBT_CLASSIFICATION_MODEL_H__
24 
25 #include "algorithms/classifier/classifier_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 classification
44 {
48 namespace interface1
49 {
62 class DAAL_EXPORT Model : public classifier::Model
63 {
64 public:
65  DECLARE_MODEL(Model, classifier::Model)
66 
67 
72  static services::SharedPtr<Model> create(size_t nFeatures, services::Status *stat = NULL);
73 
79  virtual size_t numberOfTrees() const = 0;
80 
87  virtual void traverseDF(size_t iTree, regression::TreeNodeVisitor& visitor) const = 0;
88 
95  virtual void traverseBF(size_t iTree, regression::TreeNodeVisitor& visitor) const = 0;
96 
100  virtual void clear() = 0;
101 
107  virtual void traverseDFS(size_t iTree, tree_utils::regression::TreeNodeVisitor& visitor) const = 0;
108 
114  virtual void traverseBFS(size_t iTree, tree_utils::regression::TreeNodeVisitor& visitor) const = 0;
115 
120  virtual size_t getNumberOfTrees() const = 0;
121 
122 protected:
123  Model() : classifier::Model()
124  {}
125 };
127 typedef services::SharedPtr<Model> ModelPtr;
128 } // namespace interface1
129 using interface1::Model;
130 using interface1::ModelPtr;
131 
132 } // namespace classification
133 } // namespace gbt
134 } // namespace algorithms
135 } // namespace daal
136 #endif
daal
Definition: algorithm_base_common.h:31
daal::algorithms::gbt::classification::interface1::Model
Model of the classifier trained by the gbt::training::Batch algorithm.
Definition: gbt_classification_model.h:62

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