ml.dmlc.xgboost4j.java

Class XGBoostRegTree

java.lang.Object

ml.dmlc.xgboost4j.java.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, int root_id) Retrieves nodes from root to leaf and returns leaf index.
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()

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,
                        int root_id)

Retrieves nodes from root to leaf and returns leaf index.

Specified by:

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

Parameters:

feat - feature vector

root_id - starting root index

Returns:

leaf index

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