org.apache.spark.mllib.tree.impl

Class NodeIndexUpdater

Object

org.apache.spark.mllib.tree.impl.NodeIndexUpdater

All Implemented Interfaces:

java.io.Serializable, scala.Equals, scala.Product

public class NodeIndexUpdater
extends Object
implements scala.Product, scala.Serializable

:: DeveloperApi :: This is used by the node id cache to find the child id that a data point would belong to.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
NodeIndexUpdater(Split split, int nodeIndex)

Method Summary

Methods

Modifier and Type	Method and Description
int	nodeIndex()
Split	split()
int	updateNodeIndex(int[] binnedFeatures, Bin[][] bins) Determine a child node index based on the feature value and the split.

Methods inherited from class Object

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

Methods inherited from interface scala.Product

productArity, productElement, productIterator, productPrefix

Methods inherited from interface scala.Equals

canEqual, equals

Constructor Detail

NodeIndexUpdater

public NodeIndexUpdater(Split split,
                int nodeIndex)

Method Detail

split

public Split split()

nodeIndex

public int nodeIndex()

updateNodeIndex

public int updateNodeIndex(int[] binnedFeatures,
                  Bin[][] bins)

Determine a child node index based on the feature value and the split.

Parameters:

binnedFeatures - Binned feature values.

bins - Bin information to convert the bin indices to approximate feature values.

Returns:

Child node index to update to.