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
DecisionTreeModelReadWrite.SplitData(int featureIndex, double[] leftCategoriesOrThreshold, int numCategories)

Method Summary

Methods

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

productArity, productElement, productIterator, productPrefix

Methods inherited from interface scala.Equals

canEqual, equals

Constructor Detail

DecisionTreeModelReadWrite.SplitData

public DecisionTreeModelReadWrite.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()