Package biz.k11i.xgboost.tree
Class RegTree
- java.lang.Object
-
- biz.k11i.xgboost.tree.RegTree
-
- All Implemented Interfaces:
java.io.Serializable
public class RegTree extends java.lang.Object implements java.io.Serializable
Regression tree.- See Also:
- Serialized Form
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description (package private) static class
RegTree.Node
(package private) static class
RegTree.Param
Parameters.(package private) static class
RegTree.RTreeNodeStat
Statistics each node in tree.
-
Field Summary
Fields Modifier and Type Field Description private RegTree.Node[]
nodes
private RegTree.Param
param
private RegTree.RTreeNodeStat[]
stats
-
Constructor Summary
Constructors Constructor Description RegTree()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description int
getLeafIndex(FVec feat, int root_id)
Retrieves nodes from root to leaf and returns leaf index.float
getLeafValue(FVec feat, int root_id)
Retrieves nodes from root to leaf and returns leaf value.void
loadModel(ModelReader reader)
Loads model from stream.
-
-
-
Field Detail
-
param
private RegTree.Param param
-
nodes
private RegTree.Node[] nodes
-
stats
private RegTree.RTreeNodeStat[] stats
-
-
Method Detail
-
loadModel
public void loadModel(ModelReader reader) throws java.io.IOException
Loads model from stream.- Parameters:
reader
- input stream- Throws:
java.io.IOException
- If an I/O error occurs
-
getLeafIndex
public int getLeafIndex(FVec feat, int root_id)
Retrieves nodes from root to leaf and returns leaf index.- Parameters:
feat
- feature vectorroot_id
- starting root index- Returns:
- leaf index
-
getLeafValue
public 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
-
-