org.apache.spark.ml.tree

Class DecisionTreeModelReadWrite.SplitData

Object

org.apache.spark.ml.tree.DecisionTreeModelReadWrite.SplitData

All Implemented Interfaces:

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

Enclosing class:

DecisionTreeModelReadWrite

public static class DecisionTreeModelReadWrite.SplitData
extends Object
implements scala.Product, scala.Serializable

Info for a Split

param: featureIndex Index of feature split on param: leftCategoriesOrThreshold For categorical feature, set of leftCategories. For continuous feature, threshold. param: numCategories For categorical feature, number of categories. For continuous feature, -1.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
SplitData(int featureIndex, double[] leftCategoriesOrThreshold, int numCategories)

Method Summary

Modifier and Type	Method and Description
int	featureIndex()
Split	getSplit()
double[]	leftCategoriesOrThreshold()
int	numCategories()

Methods inherited from class Object

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

Methods inherited from interface scala.Product

$init$, productArity, productElement, productIterator, productPrefix

Methods inherited from interface scala.Equals

canEqual, equals

Constructor Detail

SplitData

public SplitData(int featureIndex,
                 double[] leftCategoriesOrThreshold,
                 int numCategories)

Method Detail

featureIndex

public int featureIndex()

leftCategoriesOrThreshold

public double[] leftCategoriesOrThreshold()

numCategories

public int numCategories()

getSplit

public Split getSplit()