Tree structure for multi-target model. More...

Inheritance diagram for xgboost::MultiTargetTree:

Public Member Functions
	MultiTargetTree (TreeParam const *param)

void	SetLeaf (bst_node_t nidx, linalg::VectorView< float const > weight)
	Set the weight for a leaf.

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.

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

void	SaveModel (Json *out) const override
	saves the model config to a JSON object

Static Public Member Functions
static bst_node_t constexpr	InvalidNodeId ()

Detailed Description

Tree structure for multi-target model.

Member Function Documentation

void xgboost::MultiTargetTree::LoadModel ( Json const & in )

overridevirtual

load the model from a JSON object

Parameters

in	JSON object where to load the model from

void xgboost::MultiTargetTree::SaveModel ( Json * out ) const

overridevirtual

saves the model config to a JSON object

Parameters

out	JSON container where to save the model to

The documentation for this class was generated from the following files: