adapt tree booster

2014-02-07 22:41:32 -08:00 · 2014-02-07 22:41:32 -08:00 · bed2e26019
commit bed2e26019
parent 5d052b9e14
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();
        }
    };
 };