org.apache.spark.mllib.tree.model

Class InformationGainStats

Object

org.apache.spark.mllib.tree.model.InformationGainStats

All Implemented Interfaces:

java.io.Serializable

public class InformationGainStats
extends Object
implements scala.Serializable

:: DeveloperApi :: Information gain statistics for each split

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
InformationGainStats(double gain, double impurity, double leftImpurity, double rightImpurity, Predict leftPredict, Predict rightPredict)

Method Summary

Methods

Modifier and Type	Method and Description
boolean	equals(Object o)
double	gain()
double	impurity()
static InformationGainStats	invalidInformationGainStats() An InformationGainStats object to denote that current split doesn't satisfies minimum info gain or minimum number of instances per node.
double	leftImpurity()
Predict	leftPredict()
double	rightImpurity()
Predict	rightPredict()
String	toString()

Methods inherited from class Object

getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Detail

InformationGainStats

public InformationGainStats(double gain,
                    double impurity,
                    double leftImpurity,
                    double rightImpurity,
                    Predict leftPredict,
                    Predict rightPredict)

Method Detail

invalidInformationGainStats

public static InformationGainStats invalidInformationGainStats()

An InformationGainStats object to denote that current split doesn't satisfies minimum info gain or minimum number of instances per node.

gain

public double gain()

impurity

public double impurity()

leftImpurity

public double leftImpurity()

rightImpurity

public double rightImpurity()

leftPredict

public Predict leftPredict()

rightPredict

public Predict rightPredict()

toString

public String toString()

Overrides:

toString in class Object

equals

public boolean equals(Object o)

Overrides:

equals in class Object