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 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

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

Method Summary

boolean	equals(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()
String	toString()

Methods inherited from class Object

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

Overrides:

toString in class Object

equals

public boolean equals(Object o)

Overrides:

equals in class Object

hashCode

public int hashCode()

Overrides:

hashCode in class Object