Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add linear booster

Browse Source
This commit is contained in:
tqchen
2014-02-08 11:24:35 -08:00
parent d656d9df2c
commit 33acaaa3ae
6 changed files with 306 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
booster/xgboost.h
View File

@@ -51,7 +51,8 @@ namespace xgboost{
virtual void InitModel( void ) = 0;
public:
/*!
* \brief do gradient boost training for one step, using the information given
* \brief do gradient boost training for one step, using the information given,
* Note: content of grad and hess can change after DoBoost
* \param grad first order gradient of each instance
* \param hess second order gradient of each instance
* \param feats features of each instance
@@ -64,7 +65,7 @@ namespace xgboost{
const std::vector<unsigned> &root_index ) = 0;
/*!
* \brief predict values for given sparse feature vector
* NOTE: in tree implementation, this is not threadsafe
* NOTE: in tree implementation, this is not threadsafe, used dense version to ensure threadsafety
* \param feat vector in sparse format
* \param rid root id of current instance, default = 0
* \return prediction