set is_linear_ to false when it is absent from the model file (fix #3778) (#4056)

ec4bd1e0 · shiyu1994 · GitHub · e5c3f7e7 · ec4bd1e0
Unverified Commit ec4bd1e0 authored Mar 13, 2021 by shiyu1994 Committed by GitHub Mar 13, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

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

No files found.
--- a/src/io/tree.cpp
+++ b/src/io/tree.cpp
@@ -698,6 +698,8 @@ Tree::Tree(const char* str, size_t* used_len) {
    int is_linear_int;
    Common::Atoi(key_vals["is_linear"].c_str(), &is_linear_int);
    is_linear_ = static_cast<bool>(is_linear_int);
+  } else {
+    is_linear_ = false;
  }

  if ((num_leaves_ <= 1) && !is_linear_) {