diff --git a/booster/xgboost.cpp b/booster/xgboost.cpp index 6e04b3166..5c445dabf 100644 --- 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(); } }; };