org.apache.spark.mllib.tree.model

Class InformationGainStats

java.lang.Object

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

All Implemented Interfaces:

java.io.Serializable

public class InformationGainStats
extends java.lang.Object
implements scala.Serializable

:: DeveloperApi :: Information gain statistics for each split param: gain information gain value param: impurity current node impurity param: leftImpurity left node impurity param: rightImpurity right node impurity param: leftPredict left node predict param: rightPredict right node predict

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(java.lang.Object o)
double	gain()
int	hashCode()
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()
java.lang.String	toString()

Methods inherited from class java.lang.Object

clone, finalize, getClass, 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.

Returns:

(undocumented)

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 java.lang.String toString()

Overrides:

toString in class java.lang.Object

equals

public boolean equals(java.lang.Object o)

Overrides:

equals in class java.lang.Object

hashCode

public int hashCode()

Overrides:

hashCode in class java.lang.Object