biz.k11i.xgboost.tree

Interface RegTree

All Superinterfaces:

java.io.Serializable

All Known Implementing Classes:

RegTreeImpl

public interface RegTree
extends java.io.Serializable

Regression tree.

Method Summary

Methods

Modifier and Type	Method and Description
int	getLeafIndex(FVec feat) Retrieves nodes from root to leaf and returns leaf index.
void	getLeafPath(FVec feat, java.lang.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.
RegTreeNode[]	getNodes()

Method Detail

getLeafIndex

int getLeafIndex(FVec feat)

Retrieves nodes from root to leaf and returns leaf index.

Parameters:

feat - feature vector

Returns:

leaf index

getLeafPath

void getLeafPath(FVec feat,
               java.lang.StringBuilder sb)

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

Parameters:

feat - feature vector

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

getLeafValue

float getLeafValue(FVec feat,
                 int root_id)

Retrieves nodes from root to leaf and returns leaf value.

Parameters:

feat - feature vector

root_id - starting root index

Returns:

leaf value

getNodes

RegTreeNode[] getNodes()

Returns:

Tree's nodes