xgboost
|
树节点 更多...
#include <tree_model.h>
公共成员函数 | |
XGBOOST_DEVICE | Node () |
Node (int32_t cleft, int32_t cright, int32_t parent, uint32_t split_ind, float split_cond, bool default_left) | |
XGBOOST_DEVICE int | LeftChild () const |
左子节点的索引 更多... | |
XGBOOST_DEVICE int | RightChild () const |
右子节点的索引 更多... | |
XGBOOST_DEVICE int | DefaultChild () const |
当特征缺失时的默认子节点索引 更多... | |
XGBOOST_DEVICE bst_feature_t | SplitIndex () const |
分裂条件的特征索引 更多... | |
XGBOOST_DEVICE bool | DefaultLeft () const |
当特征未知时,是否走向左子节点 更多... | |
XGBOOST_DEVICE bool | IsLeaf () const |
当前节点是否是叶子节点 更多... | |
XGBOOST_DEVICE float | LeafValue () const |
XGBOOST_DEVICE SplitCondT | SplitCond () const |
XGBOOST_DEVICE int | Parent () const |
获取节点的父节点 更多... | |
XGBOOST_DEVICE bool | IsLeftChild () const |
当前节点是否是左子节点 更多... | |
XGBOOST_DEVICE bool | IsDeleted () const |
此节点是否已删除 更多... | |
XGBOOST_DEVICE bool | IsRoot () const |
当前节点是否是根节点 更多... | |
XGBOOST_DEVICE void | SetLeftChild (int nid) |
设置左子节点 更多... | |
XGBOOST_DEVICE void | SetRightChild (int nid) |
设置右子节点 更多... | |
XGBOOST_DEVICE void | SetSplit (unsigned split_index, SplitCondT split_cond, bool default_left=false) |
设置当前节点的分裂条件 更多... | |
XGBOOST_DEVICE void | SetLeaf (bst_float value, int right=kInvalidNodeId) |
设置节点的叶子值 更多... | |
XGBOOST_DEVICE void | MarkDelete () |
标记此节点已删除 更多... | |
XGBOOST_DEVICE void | Reuse () |
重用此已删除的节点。 更多... | |
XGBOOST_DEVICE void | SetParent (int pidx, bool is_left_child=true) |
bool | operator== (const Node &b) const |
Node | ByteSwap () const |
树节点
|
inline |
|
inline |
|
inline |
|
inline |
当特征缺失时的默认子节点索引
|
inline |
当特征未知时,是否走向左子节点
|
inline |
此节点是否已删除
|
inline |
当前节点是否是叶子节点
|
inline |
当前节点是否是左子节点
|
inline |
当前节点是否是根节点
|
inline |
|
inline |
左子节点的索引
|
inline |
标记此节点已删除
|
inline |
|
inline |
获取节点的父节点
|
inline |
重用此已删除的节点。
|
inline |
右子节点的索引
|
inline |
设置节点的叶子值
value | 叶子值 |
right | 右索引,可用于存储额外信息 |
|
inline |
设置左子节点
nid | 节点ID |
|
inline |
|
inline |
设置右子节点
nid | 节点ID |
|
inline |
设置当前节点的分裂条件
split_index | 分裂的特征索引 |
split_cond | 分裂条件 |
default_left | 特征未知时的默认方向 |
|
inline |
|
inline |
分裂条件的特征索引