Unverified Commit dace0508 authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

consider max_depth for histogram_pool_size (#2216)

* consider max_depth for histogram_pool_size

* Update serial_tree_learner.cpp

* Update config.cpp

* Update serial_tree_learner.cpp
parent b6f65783
...@@ -265,6 +265,7 @@ void Config::CheckParamConflict() { ...@@ -265,6 +265,7 @@ void Config::CheckParamConflict() {
&& num_leaves == kDefaultNumLeaves) { && num_leaves == kDefaultNumLeaves) {
Log::Warning("Accuracy may be bad since you didn't set num_leaves and 2^max_depth > num_leaves"); Log::Warning("Accuracy may be bad since you didn't set num_leaves and 2^max_depth > num_leaves");
} }
num_leaves = std::min(num_leaves, 2 << max_depth);
} }
} }
......
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