From ea4441775405b97d99ebc6b997b726b64564723b Mon Sep 17 00:00:00 2001 From: Philip Hyunsu Cho Date: Fri, 31 May 2019 12:29:23 -0700 Subject: [PATCH] Enforce exclusion between pred_interactions=True and pred_interactions=True (#4522) --- src/learner.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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);