fix #1994

b7a61c32 · Guolin Ke · 90127b52 · b7a61c32
Commit b7a61c32 authored Feb 03, 2019 by Guolin Ke
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

src/objective/regression_objective.hpp src/objective/regression_objective.hpp +5 -1

No files found.
--- a/src/objective/regression_objective.hpp
+++ b/src/objective/regression_objective.hpp
@@ -58,7 +58,11 @@ namespace LightGBM {
  CHECK(threshold < weighted_cdf[pos]);\
  T v1 = data_reader(sorted_idx[pos - 1]);\
  T v2 = data_reader(sorted_idx[pos]);\
-  return static_cast<T>((threshold - weighted_cdf[pos]) / (weighted_cdf[pos + 1] - weighted_cdf[pos]) * (v2 - v1) + v1);\
+  if(weighted_cdf[pos + 1] - weighted_cdf[pos] > kEpsilon){\
+    return static_cast<T>((threshold - weighted_cdf[pos]) / (weighted_cdf[pos + 1] - weighted_cdf[pos]) * (v2 - v1) + v1); \
+  } else {\
+    return static_cast<T>(v2);\
+  }\
 }\

 /*!