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();
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<float> preds;
if( !silent ) printf("start prediction...\n");
learner.Predict( preds, data );