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
microsoft · Jun 5, 2019 · dace050 · dace050
1 parent b6f6578
commit dace050
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/io/config.cpp b/src/io/config.cpp
@@ -265,6 +265,7 @@ void Config::CheckParamConflict() {
         && num_leaves == kDefaultNumLeaves) {
       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);
   }
 }