From f204dd7fcf388ba366834e67b8e221a44985a7ed Mon Sep 17 00:00:00 2001 From: tqchen Date: Sat, 15 Feb 2014 19:42:02 -0800 Subject: [PATCH] fix nboosters --- booster/xgboost_gbmbase.h | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/booster/xgboost_gbmbase.h b/booster/xgboost_gbmbase.h index 7ce8d844d..382836c48 100644 --- a/booster/xgboost_gbmbase.h +++ b/booster/xgboost_gbmbase.h @@ -272,10 +272,11 @@ namespace xgboost{ */ inline booster::IBooster *GetUpdateBooster( void ){ if( param.do_reboost == 0 || boosters.size() == 0 ){ + param.num_boosters += 1; boosters.push_back( booster::CreateBooster( param.booster_type ) ); booster_info.push_back( 0 ); this->ConfigBooster( boosters.back() ); - boosters.back()->InitModel(); + boosters.back()->InitModel(); }else{ this->ConfigBooster( boosters.back() ); }