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

auto do reboost

Browse Source
This commit is contained in:
tqchen 2014-03-02 16:42:22 -08:00 committed by tqchen
parent 0fc64d1c2a
commit 8ae1d37828
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
booster/xgboost_gbmbase.h
View File

@ -81,7 +81,11 @@ namespace xgboost{
* \param val value of the parameter * \param val value of the parameter
*/ */
inline void SetParam( const char *name, const char *val ){ inline void SetParam( const char *name, const char *val ){
if( !strcmp("booster_type", name ) ) booster_type = atoi( val ); if( !strcmp("booster_type", name ) ){
booster_type = atoi( val );
// linear boost automatically set do reboost
if( booster_type == 1 ) do_reboost = 1;
}
if( !strcmp("num_pbuffer", name ) ) num_pbuffer = atoi( val ); if( !strcmp("num_pbuffer", name ) ) num_pbuffer = atoi( val );
if( !strcmp("do_reboost", name ) ) do_reboost = atoi( val ); if( !strcmp("do_reboost", name ) ) do_reboost = atoi( val );
if( !strcmp("bst:num_roots", name ) ) num_roots = atoi( val ); if( !strcmp("bst:num_roots", name ) ) num_roots = atoi( val );