[TREE] Enable updater registry

2016-01-01 03:32:40 -08:00
parent a62a66d545
commit c8ccb61b9e
13 changed files with 172 additions and 189 deletions
--- a/src/tree/tree_model.cc
+++ b/src/tree/tree_model.cc
@@ -6,8 +6,9 @@
 #include <xgboost/tree_model.h>
 #include <sstream>

-
 namespace xgboost {
+// register tree parameter
+DMLC_REGISTER_PARAMETER(TreeParam);

 // internal function to dump regression tree to text
 void DumpRegTree2Text(std::stringstream& fo,  // NOLINT(*)