fix error that sum_hessian may be zero.

6c736da9 · Guolin Ke · a4a0235d · 6c736da9
Commit 6c736da9 authored Jan 23, 2017 by Guolin Ke
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/treelearner/feature_histogram.hpp src/treelearner/feature_histogram.hpp +1 -1

No files found.
--- a/src/treelearner/feature_histogram.hpp
+++ b/src/treelearner/feature_histogram.hpp
@@ -64,7 +64,7 @@ public:
  */
  void FindBestThreshold(double sum_gradient, double sum_hessian, data_size_t num_data,
    SplitInfo* output) {
-    find_best_threshold_fun_(sum_gradient, sum_hessian, num_data, output);
+    find_best_threshold_fun_(sum_gradient, sum_hessian + 2 * kEpsilon, num_data, output);
    if (output->gain > kMinScore) {
      is_splittable_ = true;
    } else {