fix load bug

2014-10-16 21:47:01 -07:00
parent 3f3c90c3c0
commit f6d61f02f6
5 changed files with 18 additions and 2 deletions
--- a/src/tree/model.h
+++ b/src/tree/model.h
@@ -305,7 +305,7 @@ class TreeModel {
    // chg deleted nodes
    deleted_nodes.resize(0);
    for (int i = param.num_roots; i < param.num_nodes; i ++) {
-      if (nodes[i].is_root()) deleted_nodes.push_back(i);
+      if (nodes[i].is_deleted()) deleted_nodes.push_back(i);
    }
    utils::Assert(static_cast<int>(deleted_nodes.size()) == param.num_deleted,
                  "number of deleted nodes do not match");