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

allow pred to stdout

Browse Source
This commit is contained in:
tqchen 2015-01-10 23:46:29 -08:00
parent db4637b085
commit 69e079941e
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/xgboost_main.cpp
View File

@ -32,7 +32,7 @@ class BoostLearnTask {
} }
} }
// do not save anything when save to stdout // do not save anything when save to stdout
if (model_out == "stdout") { if (model_out == "stdout" || name_pred == "stdout") {
this->SetParam("silent", "1"); this->SetParam("silent", "1");
save_period = 0; save_period = 0;
} }
@ -236,11 +236,16 @@ class BoostLearnTask {
if (!silent) printf("start prediction...\n"); if (!silent) printf("start prediction...\n");
learner.Predict(*data, pred_margin != 0, &preds, ntree_limit); learner.Predict(*data, pred_margin != 0, &preds, ntree_limit);
if (!silent) printf("writing prediction to %s\n", name_pred.c_str()); if (!silent) printf("writing prediction to %s\n", name_pred.c_str());
FILE *fo = utils::FopenCheck(name_pred.c_str(), "w"); FILE *fo;
for (size_t i = 0; i < preds.size(); i++) { if (name_pred != "stdout") {
fprintf(fo, "%f\n", preds[i]); fo = utils::FopenCheck(name_pred.c_str(), "w");
} else {
fo = stdout;
} }
fclose(fo); for (size_t i = 0; i < preds.size(); ++i) {
fprintf(fo, "%g\n", preds[i]);
}
if (fo != stdout) fclose(fo);
} }
private: private:
/*! \brief whether silent */ /*! \brief whether silent */