Package org.apache.spark.ml.tree

Class CategoricalSplit

Object

org.apache.spark.ml.tree.CategoricalSplit

All Implemented Interfaces:

Serializable, Split, scala.Serializable

public class CategoricalSplit
extends Object
implements Split

Split which tests a categorical feature. param: featureIndex Index of the feature to test param: _leftCategories If the feature value is in this set of categories, then the split goes left. Otherwise, it goes right. param: numCategories Number of categories for this feature.

See Also:

• Serialized Form

Method Summary

Modifier and Type	Method	Description
boolean	equals(Object o)
int	featureIndex()	Index of feature which this split tests
int	hashCode()
double[]	leftCategories()	Get sorted categories which split to the left
int	numCategories()
double[]	rightCategories()	Get sorted categories which split to the right

Methods inherited from class java.lang.Object

getClass, notify, notifyAll, toString, wait, wait, wait

Method Details

equals

public boolean equals(Object o)

Overrides:

equals in class Object

featureIndex

public int featureIndex()

Description copied from interface: Split

Index of feature which this split tests

Specified by:

featureIndex in interface Split

hashCode

public int hashCode()

Overrides:

hashCode in class Object

leftCategories

public double[] leftCategories()

Get sorted categories which split to the left

numCategories

public int numCategories()

rightCategories

public double[] rightCategories()

Get sorted categories which split to the right