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

add remove tree

Browse Source
This commit is contained in:
tqchen
2014-03-11 11:25:50 -07:00
parent d2377b26bd
commit 8f16ef8e75
5 changed files with 45 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
regression/xgboost_reg.h
View File

@@ -182,8 +182,13 @@ namespace xgboost{
* \brief update the model for one iteration
* \param iteration iteration number
*/
inline void UpdateInteract( void ){
this->InteractPredict( preds_, *train_, 0 );
inline void UpdateInteract( std::string action ){
this->InteractPredict( preds_, *train_, 0 );
if( action == "remove" ){
base_gbm.DelteBooster(); return;
}
int buffer_offset = static_cast<int>( train_->Size() );
for( size_t i = 0; i < evals_.size(); ++i ){
std::vector<float> &preds = this->eval_preds_[ i ];

8
regression/xgboost_reg_main.cpp
View File

@@ -71,6 +71,7 @@ namespace xgboost{
if( !strcmp("name_dumppath", name ) ) name_dumppath = val;
if( !strcmp("name_pred", name ) ) name_pred = val;
if( !strcmp("dump_stats", name ) ) dump_model_stats = atoi( val );
if( !strcmp("interact:action", name ) ) interact_action = val;
if( !strncmp("eval[", name, 5 ) ) {
char evname[ 256 ];
utils::Assert( sscanf( name, "eval[%[^]]", evname ) == 1, "must specify evaluation name for display");
@@ -95,6 +96,7 @@ namespace xgboost{
name_dump = "dump.txt";
name_dumppath = "dump.path.txt";
model_dir_path = "./";
interact_action = "update";
}
~RegBoostTask( void ){
for( size_t i = 0; i < deval.size(); i ++ ){
@@ -162,7 +164,7 @@ namespace xgboost{
inline void TaskInteractive( void ){
const time_t start = time( NULL );
unsigned long elapsed = 0;
learner.UpdateInteract();
learner.UpdateInteract( interact_action );
utils::Assert( model_out != "NULL", "interactive mode must specify model_out" );
this->SaveModel( model_out.c_str() );
elapsed = (unsigned long)(time(NULL) - start);
@@ -211,7 +213,9 @@ namespace xgboost{
/* \brief number of boosting iterations */
int num_round;
/* \brief the period to save the model, 0 means only save the final round model */
int save_period;
int save_period;
/*! \brief interfact action */
std::string interact_action;
/* \brief the path of training/test data set */
std::string train_path, test_path;
/* \brief the path of test model file, or file to restart training */