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

change test task to pred

Browse Source
This commit is contained in:
tqchen 2014-03-02 16:20:42 -08:00 committed by tqchen
parent a8f69878eb
commit c7b29774c2
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
regression/xgboost_reg_main.cpp
View File

@ -42,8 +42,8 @@ namespace xgboost{
this->TaskDumpPath(); this->TaskDumpPath();
return 0; return 0;
} }
if( task == "test" ){ if( task == "pred" ){
this->TaskTest(); this->TaskPred();
}else{ }else{
this->TaskTrain(); this->TaskTrain();
} }
@ -157,7 +157,7 @@ namespace xgboost{
learner.SaveModel( fo ); learner.SaveModel( fo );
fo.Close(); fo.Close();
} }
inline void TaskTest( void ){ inline void TaskPred( void ){
std::vector<float> preds; std::vector<float> preds;
if( !silent ) printf("start prediction...\n"); if( !silent ) printf("start prediction...\n");
learner.Predict( preds, data ); learner.Predict( preds, data );