Reduce base margin to 2 dim for now. (#7455)

2021-11-27 00:46:13 +08:00
parent bf7bb575b4
commit 557ffc4bf5
7 changed files with 33 additions and 33 deletions
--- a/include/xgboost/data.h
+++ b/include/xgboost/data.h
@@ -69,7 +69,7 @@ class MetaInfo {
   * if specified, xgboost will start from this init margin
   * can be used to specify initial prediction to boost from.
   */
-  linalg::Tensor<float, 3> base_margin_;  // NOLINT
+  linalg::Tensor<float, 2> base_margin_;  // NOLINT
  /*!
   * \brief lower bound of the label, to be used for survival analysis (censored regression)
   */