sync everything
This commit is contained in:
parent
a81ea03022
commit
790c76e814
@ -18,3 +18,8 @@ Planned key components (TODO):
|
|||||||
- matrix factorization
|
- matrix factorization
|
||||||
- structured prediction
|
- structured prediction
|
||||||
(3) OpenMP support for parallelization(optional)
|
(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 <vector>
|
||||||
#include "../utils/xgboost_utils.h"
|
#include "../utils/xgboost_utils.h"
|
||||||
#include "../utils/xgboost_stream.h"
|
#include "../utils/xgboost_stream.h"
|
||||||
|
#include "../utils/xgboost_config.h"
|
||||||
#include "xgboost_data.h"
|
#include "xgboost_data.h"
|
||||||
|
|
||||||
/*! \brief namespace for xboost package */
|
/*! \brief namespace for xboost package */
|
||||||
@ -44,8 +45,8 @@ namespace xgboost{
|
|||||||
virtual void SaveModel( utils::IStream &fo ) const = 0;
|
virtual void SaveModel( utils::IStream &fo ) const = 0;
|
||||||
/*!
|
/*!
|
||||||
* \brief initialize solver before training, called before training
|
* \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;
|
virtual void InitModel( void ) = 0;
|
||||||
public:
|
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
|
#endif
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user