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

change dump stats

Browse Source
This commit is contained in:
tqchen 2015-01-04 02:35:24 -08:00
parent 61df646eed
commit 2925236fab
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/tree/model.h
View File

@ -437,7 +437,6 @@ class TreeModel {
<< ",missing=" << nodes[nid].cdefault();
}
if (with_stats) {
fo << ' ';
stat(nid).Print(fo, false);
}
fo << '\n';
@ -460,9 +459,9 @@ struct RTreeNodeStat {
/*! \brief print information of current stats to fo */
inline void Print(std::stringstream &fo, bool is_leaf) const {
if (!is_leaf) {
fo << "gain=" << loss_chg << ",cover=" << sum_hess;
fo << ",gain=" << loss_chg << ",cover=" << sum_hess;
} else {
fo << "cover=" << sum_hess;
fo << ",cover=" << sum_hess;
}
}
};