org.apache.spark.mllib.tree.impurity

Interface Impurity

All Superinterfaces:

java.io.Serializable

All Known Implementing Classes:

Entropy, Gini, Variance

public interface Impurity
extends scala.Serializable

:: Experimental :: Trait for calculating information gain.

Method Summary

Methods

Modifier and Type	Method and Description
double	calculate(double c0, double c1) :: DeveloperApi :: information calculation for binary classification
double	calculate(double count, double sum, double sumSquares) :: DeveloperApi :: information calculation for regression

Method Detail

calculate

double calculate(double c0,
               double c1)

:: DeveloperApi :: information calculation for binary classification

Parameters:

c0 - count of instances with label 0

c1 - count of instances with label 1

Returns:

information value

calculate

double calculate(double count,
               double sum,
               double sumSquares)

:: DeveloperApi :: information calculation for regression

Parameters:

count - number of instances

sum - sum of labels

sumSquares - summation of squares of the labels

Returns:

information value