From c7b29774c20661bc54c5f3b887959e6a0fe64a4c Mon Sep 17 00:00:00 2001 From: tqchen Date: Sun, 2 Mar 2014 16:20:42 -0800 Subject: [PATCH] change test task to pred --- regression/xgboost_reg_main.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/regression/xgboost_reg_main.cpp b/regression/xgboost_reg_main.cpp index 298f22530..656980acb 100644 --- a/regression/xgboost_reg_main.cpp +++ b/regression/xgboost_reg_main.cpp @@ -42,8 +42,8 @@ namespace xgboost{ this->TaskDumpPath(); return 0; } - if( task == "test" ){ - this->TaskTest(); + if( task == "pred" ){ + this->TaskPred(); }else{ this->TaskTrain(); } @@ -157,7 +157,7 @@ namespace xgboost{ learner.SaveModel( fo ); fo.Close(); } - inline void TaskTest( void ){ + inline void TaskPred( void ){ std::vector preds; if( !silent ) printf("start prediction...\n"); learner.Predict( preds, data );