fix GatherInfoForThresholdNumerical boundary (fix #4286) (#4322)

346f8839 · shiyu1994 · GitHub · da3465cb · 346f8839
Unverified Commit 346f8839 authored May 26, 2021 by shiyu1994 Committed by GitHub May 26, 2021
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
@@ -572,7 +572,7 @@ class FeatureHistogram {
    const double cnt_factor = num_data / sum_hessian;
    // from right to left, and we don't need data in bin0
    for (; t >= t_end; --t) {
-      if (static_cast<uint32_t>(t + offset) < threshold) {
+      if (static_cast<uint32_t>(t + offset) <= threshold) {
        break;
      }