fix a bug in WeightedPercentile function.

0f224c05 · Guolin Ke · 2204e45f · 0f224c05
Commit 0f224c05 authored Nov 06, 2018 by Guolin Ke
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

src/objective/regression_objective.hpp src/objective/regression_objective.hpp +2 -2

No files found.
--- a/src/objective/regression_objective.hpp
+++ b/src/objective/regression_objective.hpp
@@ -48,8 +48,8 @@ namespace LightGBM {
  }\
  double threshold = weighted_cdf[cnt_data - 1] * alpha;\
  size_t pos = std::upper_bound(weighted_cdf.begin(), weighted_cdf.end(), threshold) - weighted_cdf.begin();\
-  if (pos == 0) {\
+  if (pos == 0 || pos == cnt_data - 1) {\
-    return data_reader(sorted_idx[0]);\
+    return data_reader(sorted_idx[pos]);\
  }\
  CHECK(threshold >= weighted_cdf[pos - 1]);\
  CHECK(threshold < weighted_cdf[pos]);\