diff --git a/src/learner.cc b/src/learner.cc index 709cd1af2..bf845611b 100644 --- a/src/learner.cc +++ b/src/learner.cc @@ -549,9 +549,9 @@ class LearnerImpl : public Learner { HostDeviceVector* out_preds, unsigned ntree_limit, bool pred_leaf, bool pred_contribs, bool approx_contribs, bool pred_interactions) override { - bool multiple_predictions = static_cast(pred_leaf) + - static_cast(pred_interactions) + - static_cast(pred_contribs); + int multiple_predictions = static_cast(pred_leaf) + + static_cast(pred_interactions) + + static_cast(pred_contribs); CHECK_LE(multiple_predictions, 1) << "Perform one kind of prediction at a time."; if (pred_contribs) { gbm_->PredictContribution(data, &out_preds->HostVector(), ntree_limit, approx_contribs);