biz.k11i.xgboost.tree

Class RegTreeImpl

java.lang.Object

biz.k11i.xgboost.tree.RegTreeImpl

All Implemented Interfaces:

RegTree, java.io.Serializable

public class RegTreeImpl
extends java.lang.Object
implements RegTree

Regression tree.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
RegTreeImpl()

Method Summary

Methods

Modifier and Type	Method and 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.

Methods inherited from class java.lang.Object

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

Constructor Detail

RegTreeImpl

public RegTreeImpl()

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.

Specified by:

getLeafIndex in interface RegTree

Parameters:

feat - feature vector

root_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.

Specified by:

getLeafValue in interface RegTree

Parameters:

feat - feature vector

root_id - starting root index

Returns:

leaf value