add tmp file

2014-09-12 15:52:39 -07:00
parent 87cc53f0cd
commit 3a0be47b1c
4 changed files with 30 additions and 11 deletions
--- a/src/tree/updater.cpp
+++ b/src/tree/updater.cpp
@@ -13,6 +13,8 @@ IUpdater* CreateUpdater(const char *name) {
  if (!strcmp(name, "prune")) return new TreePruner();
  if (!strcmp(name, "refresh")) return new TreeRefresher<GradStats>();
  if (!strcmp(name, "grow_colmaker")) return new ColMaker<GradStats>();
+  if (!strcmp(name, "grow_colmaker5")) return new ColMaker< CVGradStats<5> >();
+  if (!strcmp(name, "grow_colmaker3")) return new ColMaker< CVGradStats<3> >();
  utils::Error("unknown updater:%s", name);
  return NULL;
 }