Commit 45960910 authored by tks's avatar tks Committed by Guolin Ke
Browse files

fix comments (#334)

parent bc9d57b3
......@@ -180,12 +180,12 @@ public:
int num_leaves = 127;
int feature_fraction_seed = 2;
double feature_fraction = 1.0f;
// max cache size(unit:MB) for historical histogram. < 0 means not limit
// max cache size(unit:MB) for historical histogram. < 0 means no limit
double histogram_pool_size = -1.0f;
// max depth of tree model.
// Still grow tree by leaf-wise, but limit the max depth to avoid over-fitting
// And the max leaves will be min(num_leaves, pow(2, max_depth - 1))
// max_depth < 0 means not limit
// And the max leaves will be min(num_leaves, pow(2, max_depth))
// max_depth < 0 means no limit
int max_depth = -1;
int top_k = 20;
LIGHTGBM_EXPORT void Set(const std::unordered_map<std::string, std::string>& params) override;
......
......@@ -181,7 +181,7 @@ void OverallConfig::CheckParamConflict() {
&& boosting_config.tree_learner_type == std::string("data")) {
Log::Warning("Histogram LRU queue was enabled (histogram_pool_size=%f). Will disable this to reduce communication costs"
, boosting_config.tree_config.histogram_pool_size);
// Change pool size to -1 (not limit) when using data parallel to reduce communication costs
// Change pool size to -1 (no limit) when using data parallel to reduce communication costs
boosting_config.tree_config.histogram_pool_size = -1;
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment