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

add col maker

Browse Source
This commit is contained in:
tqchen
2014-03-01 14:00:09 -08:00
parent 1f04893784
commit 394d325078
5 changed files with 315 additions and 60 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
booster/tree/xgboost_tree_model.h
View File

@@ -329,6 +329,8 @@ namespace xgboost{
float subsample;
// whether to use layerwise aware regularization
int use_layerwise;
// number of threads to be used for tree construction, if OpenMP is enabled, if equals 0, use system default
int nthread;
/*! \brief constructor */
TreeParamTrain( void ){
learning_rate = 0.3f;
@@ -339,6 +341,7 @@ namespace xgboost{
default_direction = 0;
subsample = 1.0f;
use_layerwise = 0;
nthread = 0;
}
/*!
* \brief set parameters from outside
@@ -359,6 +362,7 @@ namespace xgboost{
if( !strcmp( name, "reg_method") ) reg_method = (float)atof( val );
if( !strcmp( name, "subsample") ) subsample = (float)atof( val );
if( !strcmp( name, "use_layerwise") ) use_layerwise = atoi( val );
if( !strcmp( name, "nthread") ) nthread = atoi( val );
if( !strcmp( name, "default_direction") ) {
if( !strcmp( val, "learn") ) default_direction = 0;
if( !strcmp( val, "left") ) default_direction = 1;