Package biz.k11i.xgboost.tree

Class RegTreeImpl

java.lang.Object

biz.k11i.xgboost.tree.RegTreeImpl

All Implemented Interfaces:

RegTree, Serializable

public class RegTreeImpl
extends Object
implements RegTree

Regression tree.

See Also:

• Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static class	RegTreeImpl.Node
(package private) static class	RegTreeImpl.Param	Parameters.

Field Summary

Fields

Modifier and Type	Field	Description
private RegTreeImpl.Node[]	nodes
private RegTreeImpl.Param	param
private RegTreeNodeStat[]	stats

Constructor Summary

Constructors

Constructor	Description
RegTreeImpl()

Method Summary

Modifier and Type	Method	Description
int	getLeafIndex(FVec feat)	Retrieves nodes from root to leaf and returns leaf index.
void	getLeafPath(FVec feat, StringBuilder sb)	Retrieves nodes from root to leaf and returns path to leaf.
float	getLeafValue(FVec feat, int root_id)	Retrieves nodes from root to leaf and returns leaf value.
RegTreeImpl.Node[]	getNodes()
RegTreeNodeStat[]	getStats()
void	loadModel(ModelReader reader)	Loads model from stream.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

param

private RegTreeImpl.Param param

nodes

private RegTreeImpl.Node[] nodes

stats

private RegTreeNodeStat[] stats

Constructor Details

RegTreeImpl

public RegTreeImpl()

Method Details

loadModel

public void loadModel(ModelReader reader)
               throws IOException

Loads model from stream.

Parameters:

reader - input stream

Throws:

IOException - If an I/O error occurs

getLeafIndex

public int getLeafIndex(FVec feat)

Retrieves nodes from root to leaf and returns leaf index.

Specified by:

getLeafIndex in interface RegTree

Parameters:

feat - feature vector

Returns:

leaf index

getLeafPath

public void getLeafPath(FVec feat,
 StringBuilder sb)

Retrieves nodes from root to leaf and returns path to leaf.

Specified by:

getLeafPath in interface RegTree

Parameters:

feat - feature vector

sb - output param, will write path path to leaf into this buffer

getLeafValue

public float getLeafValue(FVec feat,
 int root_id)

Retrieves nodes from root to leaf and returns leaf value.

Specified by:

getLeafValue in interface RegTree

Parameters:

feat - feature vector

root_id - starting root index

Returns:

leaf value

getNodes

public RegTreeImpl.Node[] getNodes()

Specified by:

getNodes in interface RegTree

Returns:

Tree's nodes

getStats

public RegTreeNodeStat[] getStats()

Specified by:

getStats in interface RegTree

Returns:

Tree's nodes stats