hex.tree.xgboost.predict

Class XGBoostRegTree

java.lang.Object

hex.tree.xgboost.predict.XGBoostRegTree

All Implemented Interfaces:

biz.k11i.xgboost.tree.RegTree, java.io.Serializable

public class XGBoostRegTree
extends java.lang.Object
implements biz.k11i.xgboost.tree.RegTree

Regression tree.

See Also:

Serialized Form

Method Summary

Modifier and Type	Method and Description
int	getLeafIndex(biz.k11i.xgboost.util.FVec feat)
void	getLeafPath(biz.k11i.xgboost.util.FVec fVec, java.lang.StringBuilder stringBuilder)
float	getLeafValue(biz.k11i.xgboost.util.FVec feat, int root_id) Retrieves nodes from root to leaf and returns leaf value.
biz.k11i.xgboost.tree.RegTreeNode[]	getNodes()
biz.k11i.xgboost.tree.RegTreeNodeStat[]	getStats()

Methods inherited from class java.lang.Object

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

Method Detail

getLeafIndex

public int getLeafIndex(biz.k11i.xgboost.util.FVec feat)

Specified by:

getLeafIndex in interface biz.k11i.xgboost.tree.RegTree

getLeafPath

public void getLeafPath(biz.k11i.xgboost.util.FVec fVec,
                        java.lang.StringBuilder stringBuilder)

Specified by:

getLeafPath in interface biz.k11i.xgboost.tree.RegTree

getLeafValue

public final float getLeafValue(biz.k11i.xgboost.util.FVec feat,
                                int root_id)

Retrieves nodes from root to leaf and returns leaf value.

Specified by:

getLeafValue in interface biz.k11i.xgboost.tree.RegTree

Parameters:

feat - feature vector

root_id - starting root index

Returns:

leaf value

getNodes

public biz.k11i.xgboost.tree.RegTreeNode[] getNodes()

Specified by:

getNodes in interface biz.k11i.xgboost.tree.RegTree

getStats

public biz.k11i.xgboost.tree.RegTreeNodeStat[] getStats()

Specified by:

getStats in interface biz.k11i.xgboost.tree.RegTree