CSR-like matrix for categorical splits. More...
#include <tree_model.h>
Collaboration diagram for xgboost::RegTree::CategoricalSplitMatrix:
Classes | |
| struct | Segment |
Public Attributes | |
| common::Span< FeatureType const > | split_type |
| common::Span< uint32_t const > | categories |
| common::Span< Segment const > | node_ptr |
Detailed Description
CSR-like matrix for categorical splits.
The fields of split_categories_segments_[i] are set such that the range node_ptr[beg:(beg+size)] stores the bitset for the matching categories for the i-th node.
Member Data Documentation
◆ categories
| common::Span<uint32_t const> xgboost::RegTree::CategoricalSplitMatrix::categories |
◆ node_ptr
| common::Span<Segment const> xgboost::RegTree::CategoricalSplitMatrix::node_ptr |
◆ split_type
| common::Span<FeatureType const> xgboost::RegTree::CategoricalSplitMatrix::split_type |
The documentation for this struct was generated from the following file:
- /home/docs/checkouts/readthedocs.org/user_builds/xgboost/checkouts/latest/include/xgboost/tree_model.h
