#ifndef XGBOOST_INL_HPP #define XGBOOST_INL_HPP /*! * \file xgboost-inl.hpp * \brief bootser implementations * \author Tianqi Chen: tianqi.tchen@gmail.com */ // implementation of boosters go to here #include "xgboost.h" #include "../utils/xgboost_utils.h" #include "tree/xgboost_tree.hpp" #include "linear/xgboost_linear.hpp" namespace xgboost{ namespace booster{ /*! * \brief create a gradient booster, given type of booster * \param booster_type type of gradient booster, can be used to specify implements * \tparam FMatrix input data type for booster * \return the pointer to the gradient booster created */ template inline InterfaceBooster *CreateBooster( int booster_type ){ switch( booster_type ){ case 0: return new RegTreeTrainer(); case 1: return new LinearBooster(); default: utils::Error("unknown booster_type"); return NULL; } } }; }; #endif // XGBOOST_INL_HPP