Support exporting cut values (#9356)

2023-07-08 15:32:41 +08:00
parent c3124813e8
commit 20c52f07d2
28 changed files with 722 additions and 101 deletions
--- a/src/common/api_entry.h
+++ b/src/common/api_entry.h
@@ -24,6 +24,8 @@ struct XGBAPIThreadLocalEntry {
  std::vector<const char *> ret_vec_charp;
  /*! \brief returning float vector. */
  std::vector<float> ret_vec_float;
+  /*! \brief returning uint vector. */
+  std::vector<std::uint64_t> ret_vec_u64;
  /*! \brief temp variable of gradient pairs. */
  std::vector<GradientPair> tmp_gpair;
  /*! \brief Temp variable for returning prediction result. */