Medial Code Documentation
|
construct a tree using quantized feature values More...
Public Member Functions | |
QuantileHistMaker (Context const *ctx, ObjInfo const *task) | |
void | Configure (Args const &args) override |
Initialize the updater with given arguments. | |
void | LoadConfig (Json const &in) override |
Load configuration from JSON object. | |
void | SaveConfig (Json *p_out) const override |
Save configuration to JSON object. | |
char const * | Name () const override |
void | Update (TrainParam const *param, HostDeviceVector< GradientPair > *gpair, DMatrix *p_fmat, common::Span< HostDeviceVector< bst_node_t > > out_position, const std::vector< RegTree * > &trees) override |
perform update to the tree models | |
bool | UpdatePredictionCache (const DMatrix *data, linalg::MatrixView< float > out_preds) override |
determines whether updater has enough knowledge about a given dataset to quickly update prediction cache its training data and performs the update if possible. | |
bool | HasNodePosition () const override |
Wether the out_position in Update is valid. This determines whether adaptive tree can be used. | |
![]() | |
TreeUpdater (const Context *ctx) | |
~TreeUpdater () override=default | |
virtual destructor | |
virtual bool | CanModifyTree () const |
Whether this updater can be used for updating existing trees. | |
Additional Inherited Members | |
![]() | |
static TreeUpdater * | Create (const std::string &name, Context const *ctx, ObjInfo const *task) |
Create a tree updater given name. | |
![]() | |
Context const * | ctx_ = nullptr |
construct a tree using quantized feature values
|
inlineoverridevirtual |
Initialize the updater with given arguments.
args | arguments to the objective function. |
Implements xgboost::TreeUpdater.
|
inlineoverridevirtual |
Wether the out_position in Update
is valid. This determines whether adaptive tree can be used.
Reimplemented from xgboost::TreeUpdater.
|
inlineoverridevirtual |
Load configuration from JSON object.
in | JSON object containing the configuration |
Implements xgboost::Configurable.
|
inlineoverridevirtual |
Implements xgboost::TreeUpdater.
|
inlineoverridevirtual |
Save configuration to JSON object.
out | pointer to output JSON object |
Implements xgboost::Configurable.
|
inlineoverridevirtual |
perform update to the tree models
param | Hyper-parameter for constructing trees. |
gpair | the gradient pair statistics of the data |
data | The data matrix passed to the updater. |
out_position | The leaf index for each row. The index is negated if that row is removed during sampling. So the 3th node is ~3. |
out_trees | references the trees to be updated, updater will change the content of trees note: all the trees in the vector are updated, with the same statistics, but maybe different random seeds, usually one tree is passed in at a time, there can be multiple trees when we train random forest style model |
Implements xgboost::TreeUpdater.
|
inlineoverridevirtual |
determines whether updater has enough knowledge about a given dataset to quickly update prediction cache its training data and performs the update if possible.
data | data matrix |
out_preds | prediction cache to be updated |
Reimplemented from xgboost::TreeUpdater.