fix #1994

microsoft · Feb 3, 2019 · b7a61c3 · b7a61c3
1 parent 90127b5
commit b7a61c3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git 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);\
+  }\
 }\
 
 /*!