fix dump tree (#465)

8fd7e185 · wxchan · Guolin Ke · 567f63ec · 8fd7e185
Commit 8fd7e185 authored Apr 27, 2017 by wxchan Committed by Guolin Ke Apr 27, 2017
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 1 deletion

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

No files found.
--- a/src/io/tree.cpp
+++ b/src/io/tree.cpp
@@ -329,7 +329,11 @@ std::string Tree::ToJSON() {
  str_buf << "\"num_leaves\":" << num_leaves_ << "," << std::endl;
  str_buf << "\"shrinkage\":" << shrinkage_ << "," << std::endl;
  str_buf << "\"has_categorical\":" << (has_categorical_ ? 1 : 0) << "," << std::endl;
+  if (num_leaves_ == 1) {
+    str_buf << "\"tree_structure\":" << NodeToJSON(-1) << std::endl;
+  } else {
    str_buf << "\"tree_structure\":" << NodeToJSON(0) << std::endl;
+  }

  return str_buf.str();
 }