Package biz.k11i.xgboost.tree
Class RegTree
java.lang.Object
biz.k11i.xgboost.tree.RegTree
- All Implemented Interfaces:
Serializable
Regression tree.
- See Also:
-
Nested Class Summary
Nested ClassesModifier and TypeClassDescription(package private) static class
(package private) static class
Parameters.(package private) static class
Statistics each node in tree. -
Field Summary
FieldsModifier and TypeFieldDescriptionprivate RegTree.Node[]
private RegTree.Param
private RegTree.RTreeNodeStat[]
-
Constructor Summary
Constructors -
Method Summary
Modifier and TypeMethodDescriptionint
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 Details
-
param
-
nodes
-
stats
-
-
Constructor Details
-
RegTree
public RegTree()
-
-
Method Details
-
loadModel
Loads model from stream.- Parameters:
reader
- input stream- Throws:
IOException
- If an I/O error occurs
-
getLeafIndex
Retrieves nodes from root to leaf and returns leaf index.- Parameters:
feat
- feature vectorroot_id
- starting root index- Returns:
- leaf index
-
getLeafValue
Retrieves nodes from root to leaf and returns leaf value.- Parameters:
feat
- feature vectorroot_id
- starting root index- Returns:
- leaf value
-