xgboost
|
Tree structure for multi-target model. More...
#include <multi_target_tree_model.h>
Public Member Functions | |
MultiTargetTree (TreeParam const *param) | |
void | SetLeaf (bst_node_t nidx, linalg::VectorView< float const > weight) |
Set the weight for a leaf. More... | |
void | Expand (bst_node_t nidx, bst_feature_t split_idx, float split_cond, bool default_left, linalg::VectorView< float const > base_weight, linalg::VectorView< float const > left_weight, linalg::VectorView< float const > right_weight) |
Expand a leaf into split node. More... | |
bool | IsLeaf (bst_node_t nidx) const |
bst_node_t | Parent (bst_node_t nidx) const |
bst_node_t | LeftChild (bst_node_t nidx) const |
bst_node_t | RightChild (bst_node_t nidx) const |
bst_feature_t | SplitIndex (bst_node_t nidx) const |
float | SplitCond (bst_node_t nidx) const |
bool | DefaultLeft (bst_node_t nidx) const |
bst_node_t | DefaultChild (bst_node_t nidx) const |
bst_target_t | NumTarget () const |
std::size_t | Size () const |
bst_node_t | Depth (bst_node_t nidx) const |
linalg::VectorView< float const > | LeafValue (bst_node_t nidx) const |
void | LoadModel (Json const &in) override |
load the model from a JSON object More... | |
void | SaveModel (Json *out) const override |
saves the model config to a JSON object More... | |
Public Member Functions inherited from xgboost::Model | |
virtual | ~Model ()=default |
Static Public Member Functions | |
static constexpr bst_node_t | InvalidNodeId () |
Tree structure for multi-target model.
|
explicit |
|
inline |
|
inline |
|
inline |
void xgboost::MultiTargetTree::Expand | ( | bst_node_t | nidx, |
bst_feature_t | split_idx, | ||
float | split_cond, | ||
bool | default_left, | ||
linalg::VectorView< float const > | base_weight, | ||
linalg::VectorView< float const > | left_weight, | ||
linalg::VectorView< float const > | right_weight | ||
) |
Expand a leaf into split node.
|
inlinestaticconstexpr |
|
inline |
|
inline |
|
inline |
|
overridevirtual |
load the model from a JSON object
in | JSON object where to load the model from |
Implements xgboost::Model.
bst_target_t xgboost::MultiTargetTree::NumTarget | ( | ) | const |
|
inline |
|
inline |
|
overridevirtual |
saves the model config to a JSON object
out | JSON container where to save the model to |
Implements xgboost::Model.
void xgboost::MultiTargetTree::SetLeaf | ( | bst_node_t | nidx, |
linalg::VectorView< float const > | weight | ||
) |
Set the weight for a leaf.
std::size_t xgboost::MultiTargetTree::Size | ( | ) | const |
|
inline |
|
inline |