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

Rename parameter in fast_hist to disambiguate (#1962)

Browse Source
This commit is contained in:
Philip Cho 2017-01-13 11:35:55 -08:00 committed by Tianqi Chen
parent aeb4e76118
commit 49ff7c1649
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/tree/param.h
View File

@ -38,7 +38,7 @@ struct TrainParam : public dmlc::Parameter<TrainParam> {
// growing policy // growing policy
enum TreeGrowPolicy { kDepthWise = 0, kLossGuide = 1 }; enum TreeGrowPolicy { kDepthWise = 0, kLossGuide = 1 };
int grow_policy; int grow_policy;
int verbose; int debug_verbose;
//----- the rest parameters are less important ---- //----- the rest parameters are less important ----
// minimum amount of hessian(weight) allowed in a child // minimum amount of hessian(weight) allowed in a child
float min_child_weight; float min_child_weight;
@ -87,7 +87,7 @@ struct TrainParam : public dmlc::Parameter<TrainParam> {
.set_default(0.0f) .set_default(0.0f)
.describe( .describe(
"Minimum loss reduction required to make a further partition."); "Minimum loss reduction required to make a further partition.");
DMLC_DECLARE_FIELD(verbose) DMLC_DECLARE_FIELD(debug_verbose)
.set_lower_bound(0) .set_lower_bound(0)
.set_default(0) .set_default(0)
.describe( .describe(

4
src/tree/updater_fast_hist.cc
View File

@ -52,7 +52,7 @@ class FastHistMaker: public TreeUpdater {
gmat_.cut = &hmat_; gmat_.cut = &hmat_;
gmat_.Init(dmat); gmat_.Init(dmat);
is_gmat_initialized_ = true; is_gmat_initialized_ = true;
if (param.verbose > 0) { if (param.debug_verbose > 0) {
LOG(INFO) << "Generating gmat: " << dmlc::GetTime() - tstart << " sec"; LOG(INFO) << "Generating gmat: " << dmlc::GetTime() - tstart << " sec";
} }
} }
@ -222,7 +222,7 @@ class FastHistMaker: public TreeUpdater {
snode[nid].stats.SetLeafVec(param, p_tree->leafvec(nid)); snode[nid].stats.SetLeafVec(param, p_tree->leafvec(nid));
} }
if (param.verbose > 0) { if (param.debug_verbose > 0) {
double total_time = dmlc::GetTime() - gstart; double total_time = dmlc::GetTime() - gstart;
LOG(INFO) << "\nInitData: " LOG(INFO) << "\nInitData: "
<< std::fixed << std::setw(4) << std::setprecision(2) << time_init_data << std::fixed << std::setw(4) << std::setprecision(2) << time_init_data