Package biz.k11i.xgboost.tree
Interface RegTree
- All Superinterfaces:
Serializable
- All Known Implementing Classes:
RegTreeImpl
Regression tree.
-
Method Summary
Modifier and TypeMethodDescriptionintgetLeafIndex(FVec feat) Retrieves nodes from root to leaf and returns leaf index.voidgetLeafPath(FVec feat, StringBuilder sb) Retrieves nodes from root to leaf and returns path to leaf.floatgetLeafValue(FVec feat, int root_id) Retrieves nodes from root to leaf and returns leaf value.getNodes()getStats()
-
Method Details
-
getLeafIndex
Retrieves nodes from root to leaf and returns leaf index.- Parameters:
feat- feature vector- Returns:
- leaf index
-
getLeafPath
Retrieves nodes from root to leaf and returns path to leaf.- Parameters:
feat- feature vectorsb- output param, will write path path to leaf into this buffer
-
getLeafValue
Retrieves nodes from root to leaf and returns leaf value.- Parameters:
feat- feature vectorroot_id- starting root index- Returns:
- leaf value
-
getNodes
RegTreeNode[] getNodes()- Returns:
- Tree's nodes
-
getStats
RegTreeNodeStat[] getStats()- Returns:
- Tree's nodes stats
-