Updates to GPUTreeShap (#6087)

* Extract paths on device * Update GPUTreeShap
2020-09-06 13:39:08 +12:00
parent 0e2d5669f6
commit 2e907abdb8
4 changed files with 110 additions and 52 deletions
--- a/include/xgboost/tree_model.h
+++ b/include/xgboost/tree_model.h
@@ -337,6 +337,9 @@ class RegTree : public Model {
  /*! \brief get const reference to nodes */
  const std::vector<Node>& GetNodes() const { return nodes_; }

+  /*! \brief get const reference to stats */
+  const std::vector<RTreeNodeStat>& GetStats() const { return stats_; }
+
  /*! \brief get node statistics given nid */
  RTreeNodeStat& Stat(int nid) {
    return stats_[nid];