Fix inplace prediction interval. (#6259)

* Add back the interval in call.
* Make the interval non-optional.
This commit is contained in:
Jiaming Yuan
2020-10-28 13:13:59 +08:00
committed by GitHub
parent cc76724762
commit 3310e208fd
6 changed files with 17 additions and 12 deletions

View File

@@ -207,8 +207,8 @@ class GBTree : public GradientBooster {
void InplacePredict(dmlc::any const &x, float missing,
PredictionCacheEntry *out_preds,
uint32_t layer_begin = 0,
unsigned layer_end = 0) const override {
uint32_t layer_begin,
unsigned layer_end) const override {
CHECK(configured_);
// From here on, layer becomes concrete trees.
bst_group_t groups = model_.learner_model_param->num_output_group;