diff --git a/regression/xgboost_reg.h b/regression/xgboost_reg.h index 2aedf8672..2abe436b7 100644 --- a/regression/xgboost_reg.h +++ b/regression/xgboost_reg.h @@ -185,16 +185,16 @@ namespace xgboost{ inline void UpdateInteract( std::string action ){ this->InteractPredict( preds_, *train_, 0 ); - if( action == "remove" ){ - base_gbm.DelteBooster(); return; - } - int buffer_offset = static_cast( train_->Size() ); for( size_t i = 0; i < evals_.size(); ++i ){ std::vector &preds = this->eval_preds_[ i ]; this->InteractPredict( preds, *evals_[i], buffer_offset ); buffer_offset += static_cast( evals_[i]->Size() ); } + + if( action == "remove" ){ + base_gbm.DelteBooster(); return; + } this->GetGradient( preds_, train_->labels, grad_, hess_ ); std::vector root_index;