sync everything
This commit is contained in:
parent
51d8409e30
commit
4e2d67b81a
@ -18,3 +18,8 @@ Planned key components (TODO):
|
||||
- matrix factorization
|
||||
- structured prediction
|
||||
(3) OpenMP support for parallelization(optional)
|
||||
|
||||
File extension convention:
|
||||
(1) .h are interface, utils anddata structures, with detailed comment;
|
||||
(2) .cpp are implementations that will be compiled, with less comment;
|
||||
(3) .hpp are implementations that will be included by .cpp, with less comment
|
||||
|
||||
@ -8,6 +8,7 @@
|
||||
#include <vector>
|
||||
#include "../utils/xgboost_utils.h"
|
||||
#include "../utils/xgboost_stream.h"
|
||||
#include "../utils/xgboost_config.h"
|
||||
#include "xgboost_data.h"
|
||||
|
||||
/*! \brief namespace for xboost package */
|
||||
@ -44,8 +45,8 @@ namespace xgboost{
|
||||
virtual void SaveModel( utils::IStream &fo ) const = 0;
|
||||
/*!
|
||||
* \brief initialize solver before training, called before training
|
||||
* this function is reserved for solver to allocate necessary space and do other preparations
|
||||
*/
|
||||
* this function is reserved for solver to allocate necessary space and do other preparation
|
||||
*/
|
||||
virtual void InitModel( void ) = 0;
|
||||
public:
|
||||
/*!
|
||||
@ -96,4 +97,14 @@ namespace xgboost{
|
||||
};
|
||||
};
|
||||
|
||||
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
|
||||
* \return the pointer to the gradient booster created
|
||||
*/
|
||||
IBooster *create_booster( int booster_type );
|
||||
};
|
||||
};
|
||||
#endif
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user