xgboost
|
多目标模型的树结构。 更多...
#include <multi_target_tree_model.h>
公有成员函数 | |
MultiTargetTree (TreeParam const *param) | |
void | SetLeaf (bst_node_t nidx, linalg::VectorView< float const > weight) |
设置叶子节点的权重。 更多... | |
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) |
将叶子节点扩展为分裂节点。 更多... | |
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 |
从 JSON 对象加载模型 更多... | |
void | SaveModel (Json *out) const override |
将模型配置保存到 JSON 对象 更多... | |
![]() | |
virtual | ~Model ()=default |
静态公有成员函数 | |
static constexpr bst_node_t | InvalidNodeId () |
多目标模型的树结构。
|
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 | ||
) |
将叶子节点扩展为分裂节点。
|
inlinestaticconstexpr |
|
inline |
|
inline |
|
inline |
|
overridevirtual |
bst_target_t xgboost::MultiTargetTree::NumTarget | ( | ) | const |
|
inline |
|
inline |
|
overridevirtual |
void xgboost::MultiTargetTree::SetLeaf | ( | bst_node_t | nidx, |
linalg::VectorView< float const > | weight | ||
) |
设置叶子节点的权重。
std::size_t xgboost::MultiTargetTree::Size | ( | ) | const |
|
inline |
|
inline |