adapt tree booster

2014-02-07 22:41:32 -08:00 · 2014-02-07 22:41:32 -08:00 · 0febb1a443
commit 0febb1a443
parent 36a04f17df
1 changed files with 1 additions and 2 deletions
--- a/booster/xgboost.cpp
+++ b/booster/xgboost.cpp
@ -21,8 +21,7 @@ namespace xgboost{
         * \return the pointer to the gradient booster created
         */
        IBooster *CreateBooster( int booster_type ){
-            // TODO
-            return NULL;
+            return new RTreeTrainer();
        }
    };
 };