Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix nboosters

Browse Source
This commit is contained in:
tqchen 2014-02-15 19:42:02 -08:00 committed by tqchen
parent c38399b989
commit f204dd7fcf
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
booster/xgboost_gbmbase.h
View File

@ -272,10 +272,11 @@ namespace xgboost{
*/ */
inline booster::IBooster *GetUpdateBooster( void ){ inline booster::IBooster *GetUpdateBooster( void ){
if( param.do_reboost == 0 || boosters.size() == 0 ){ if( param.do_reboost == 0 || boosters.size() == 0 ){
param.num_boosters += 1;
boosters.push_back( booster::CreateBooster( param.booster_type ) ); boosters.push_back( booster::CreateBooster( param.booster_type ) );
booster_info.push_back( 0 ); booster_info.push_back( 0 );
this->ConfigBooster( boosters.back() ); this->ConfigBooster( boosters.back() );
boosters.back()->InitModel(); boosters.back()->InitModel();
}else{ }else{
this->ConfigBooster( boosters.back() ); this->ConfigBooster( boosters.back() );
} }