fix precision lost in tree's ToIfElse (#5187)

a107c907 · Lipson · GitHub · 0fb09e77 · a107c907
Unverified Commit a107c907 authored May 02, 2022 by Lipson Committed by GitHub May 01, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

src/io/tree.cpp src/io/tree.cpp +1 -0

No files found.
--- a/src/io/tree.cpp
+++ b/src/io/tree.cpp
@@ -524,6 +524,7 @@ std::string Tree::NodeToJSON(int index) const {
 std::string Tree::NumericalDecisionIfElse(int node) const {
  std::stringstream str_buf;
  Common::C_stringstream(str_buf);
+  str_buf << std::setprecision(std::numeric_limits<double>::digits10 + 2);
  uint8_t missing_type = GetMissingType(decision_type_[node]);
  bool default_left = GetDecisionType(decision_type_[node], kDefaultLeftMask);
  if (missing_type == MissingType::None