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
This commit is contained in:
Guolin Ke 2019-06-05 08:30:44 +08:00 коммит произвёл GitHub
Родитель b6f6578368
Коммит dace0508cb
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 1 добавлений и 0 удалений

Просмотреть файл

@ -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);
}
}