Check number of trees in inplace predict. (#7409)

2021-11-12 18:20:23 +08:00
parent 97d7582457
commit ca6f980932
2 changed files with 44 additions and 0 deletions
--- a/src/gbm/gbtree.h
+++ b/src/gbm/gbtree.h
@@ -273,6 +273,7 @@ class GBTree : public GradientBooster {
    uint32_t tree_begin, tree_end;
    std::tie(tree_begin, tree_end) =
        detail::LayerToTree(model_, tparam_, layer_begin, layer_end);
+    CHECK_LE(tree_end, model_.trees.size()) << "Invalid number of trees.";
    std::vector<Predictor const *> predictors{
      cpu_predictor_.get(),
 #if defined(XGBOOST_USE_CUDA)