Fix num_roots to be 1. (#5165)

2019-12-30 02:18:45 +08:00
parent d55489af14
commit 139ccc9902
3 changed files with 3 additions and 0 deletions
--- a/include/xgboost/tree_model.h
+++ b/include/xgboost/tree_model.h
@@ -56,6 +56,7 @@ struct TreeParam : public dmlc::Parameter<TreeParam> {
                  "TreeParam: 64 bit align");
    std::memset(this, 0, sizeof(TreeParam));
    num_nodes = 1;
+    deprecated_num_roots = 1;
  }
  // declare the parameters
  DMLC_DECLARE_PARAMETER(TreeParam) {