fix now today, try to think how to work tmr

2014-08-24 22:08:21 -07:00
parent f62b4a02f9
commit fd03239b77
2 changed files with 7 additions and 2 deletions
--- a/wrapper/xgboost_wrapper.cpp
+++ b/wrapper/xgboost_wrapper.cpp
@@ -154,6 +154,9 @@ extern "C"{
      if (src.info.info.root_index.size() != 0) {
        ret.info.info.root_index.push_back(src.info.info.root_index[ridx]);
      }
+      if (src.info.info.fold_index.size() != 0) {
+        ret.info.info.fold_index.push_back(src.info.info.fold_index[ridx]);
+      }
    }
    return p_ret;
  }