Update TreeVisitor with node cover value

oneapi-src · Oct 10, 2023 · bc04a68 · bc04a68
1 parent e418ec4
commit bc04a68
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/cpp/daal/include/algorithms/tree_utils/tree_utils.h b/cpp/daal/include/algorithms/tree_utils/tree_utils.h
@@ -59,6 +59,7 @@ struct DAAL_EXPORT SplitNodeDescriptor : public NodeDescriptor
 {
     size_t featureIndex; /*!< Feature used for splitting the node */
     double featureValue; /*!< Threshold value at the node */
+    double coverValue;   /*!< Cover (sum_hess) for the node */
 };
 
 /**

diff --git a/cpp/daal/include/algorithms/tree_utils/tree_utils_classification.h b/cpp/daal/include/algorithms/tree_utils/tree_utils_classification.h
@@ -50,8 +50,9 @@ namespace interface2
  */
 struct DAAL_EXPORT LeafNodeDescriptor : public NodeDescriptor
 {
-    size_t label;        /*!< Label to be predicted when reaching the leaf */
-    const double * prob; /*!< Probabilities estimation for the leaf */
+    size_t label;         /*!< Label to be predicted when reaching the leaf */
+    const double * prob;  /*!< Probabilities estimation for the leaf */
+    const double * cover; /*!< Cover (sum_hess) for the leaf */
 };
 
 typedef daal::algorithms::tree_utils::TreeNodeVisitor<LeafNodeDescriptor> TreeNodeVisitor;

diff --git a/cpp/daal/include/algorithms/tree_utils/tree_utils_regression.h b/cpp/daal/include/algorithms/tree_utils/tree_utils_regression.h
@@ -50,7 +50,8 @@ namespace interface1
  */
 struct DAAL_EXPORT LeafNodeDescriptor : public NodeDescriptor
 {
-    double response; /*!< Value to be predicted when reaching the leaf */
+    double response;   /*!< Value to be predicted when reaching the leaf */
+    double coverValue; /*!< Cover (sum_hess) for the leaf */
 };
 
 typedef daal::algorithms::tree_utils::TreeNodeVisitor<LeafNodeDescriptor> TreeNodeVisitor;