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 classParameters.(package private) static classStatistics each node in tree. -
Field Summary
FieldsModifier and TypeFieldDescriptionprivate RegTree.Node[]private RegTree.Paramprivate RegTree.RTreeNodeStat[] -
Constructor Summary
Constructors -
Method Summary
Modifier and TypeMethodDescriptionintgetLeafIndex(FVec feat, int root_id) Retrieves nodes from root to leaf and returns leaf index.floatgetLeafValue(FVec feat, int root_id) Retrieves nodes from root to leaf and returns leaf value.voidloadModel(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
-