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
Methods inherited from class Object |
getClass, notify, notifyAll, wait, wait, wait |
InformationGainStats
public InformationGainStats(double gain,
double impurity,
double leftImpurity,
double rightImpurity,
Predict leftPredict,
Predict rightPredict)
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