org.apache.spark.ml.tree

Class CategoricalSplit

java.lang.Object

org.apache.spark.ml.tree.CategoricalSplit

All Implemented Interfaces:

java.io.Serializable, Split

public final class CategoricalSplit
extends java.lang.Object
implements Split

:: DeveloperApi :: 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

Methods

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

Methods inherited from class java.lang.Object

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

Method Detail

featureIndex

public int featureIndex()

Description copied from interface: Split

Index of feature which this split tests

Specified by:

featureIndex in interface Split

equals

public boolean equals(java.lang.Object o)

Overrides:

equals in class java.lang.Object

leftCategories

public double[] leftCategories()

Get sorted categories which split to the left

rightCategories

public double[] rightCategories()

Get sorted categories which split to the right