org.apache.spark.mllib.tree.impurity

Class Entropy

Object

org.apache.spark.mllib.tree.impurity.Entropy

All Implemented Interfaces:

java.io.Serializable, Impurity

public class Entropy
extends Object
implements Impurity

:: Experimental :: Class for calculating entropy during binary classification.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
Entropy()

Method Summary

Methods

Modifier and Type	Method and Description
static double	calculate(double c0, double c1) :: DeveloperApi :: entropy calculation
static double	calculate(double count, double sum, double sumSquares)

Methods inherited from class Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface org.apache.spark.mllib.tree.impurity.Impurity

calculate, calculate

Constructor Detail

Entropy

public Entropy()

Method Detail

calculate

public static double calculate(double c0,
               double c1)

:: DeveloperApi :: entropy calculation

Parameters:

c0 - count of instances with label 0

c1 - count of instances with label 1

Returns:

entropy value

calculate

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