Package biz.k11i.xgboost.tree
Interface RegTree
-
- All Superinterfaces:
java.io.Serializable
- All Known Implementing Classes:
RegTreeImpl
public interface RegTree extends java.io.SerializableRegression tree.
-
-
Method Summary
All Methods Instance Methods Abstract Methods Modifier and Type Method Description intgetLeafIndex(FVec feat)Retrieves nodes from root to leaf and returns leaf index.voidgetLeafPath(FVec feat, java.lang.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.RegTreeNode[]getNodes()RegTreeNodeStat[]getStats()
-
-
-
Method Detail
-
getLeafIndex
int getLeafIndex(FVec feat)
Retrieves nodes from root to leaf and returns leaf index.- Parameters:
feat- feature vector- Returns:
- leaf index
-
getLeafPath
void getLeafPath(FVec feat, java.lang.StringBuilder sb)
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
float getLeafValue(FVec feat, int root_id)
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
-
-