org.apache.spark.ml.tree

Interface Split

All Superinterfaces:

java.io.Serializable

All Known Implementing Classes:

CategoricalSplit, ContinuousSplit

public interface Split
extends scala.Serializable

Interface for a "Split," which specifies a test made at a decision tree node to choose the left or right path.

Method Summary

Modifier and Type	Method and Description
int	featureIndex() Index of feature which this split tests
boolean	shouldGoLeft(int binnedFeature, Split[] splits) Return true (split to left) or false (split to right).
boolean	shouldGoLeft(Vector features) Return true (split to left) or false (split to right).
Split	toOld() Convert to old Split format

Method Detail

featureIndex

int featureIndex()

Index of feature which this split tests

shouldGoLeft

boolean shouldGoLeft(Vector features)

Return true (split to left) or false (split to right).

Parameters:

features - Vector of features (original values, not binned).

Returns:

(undocumented)

shouldGoLeft

boolean shouldGoLeft(int binnedFeature,
                     Split[] splits)

Return true (split to left) or false (split to right).

Parameters:

binnedFeature - Binned feature value.

splits - All splits for the given feature.

Returns:

(undocumented)

toOld

Split toOld()

Convert to old Split format