|
Medial Code Documentation
|
Public Member Functions | |
| CPUExpandEntry (bst_node_t nidx, bst_node_t depth, SplitEntry split) | |
| CPUExpandEntry (bst_node_t nidx, bst_node_t depth) | |
| bool | IsValidImpl (TrainParam const ¶m, bst_node_t num_leaves) const |
| void | CopyAndCollect (CPUExpandEntry const &that, std::vector< uint32_t > *collected_cat_bits, std::vector< std::size_t > *cat_bits_sizes) |
| Copy primitive fields into this, and collect cat_bits into a vector. | |
Public Member Functions inherited from xgboost::tree::ExpandEntryImpl< CPUExpandEntry > | |
| float | GetLossChange () const |
| bst_node_t | GetNodeId () const |
| bool | IsValid (TrainParam const ¶m, bst_node_t num_leaves) const |
Data Fields | |
| SplitEntry | split |
Data Fields inherited from xgboost::tree::ExpandEntryImpl< CPUExpandEntry > | |
| bst_node_t | nid |
| bst_node_t | depth |
Friends | |
| std::ostream & | operator<< (std::ostream &os, CPUExpandEntry const &e) |
|
inline |
Copy primitive fields into this, and collect cat_bits into a vector.
This is used for allgather.
| that | The other entry to copy from |
| collected_cat_bits | The vector to collect cat_bits |
| cat_bits_sizes | The sizes of the collected cat_bits |