org.nd4j.evaluation.classification

Class ConfusionMatrix<T extends Comparable<? super T>>

java.lang.Object

org.nd4j.evaluation.classification.ConfusionMatrix<T>

All Implemented Interfaces:

Serializable

public class ConfusionMatrix<T extends Comparable<? super T>>
extends Object
implements Serializable

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
ConfusionMatrix()
ConfusionMatrix(ConfusionMatrix<T> other) Creates a new ConfusionMatrix initialized with the contents of another ConfusionMatrix.
ConfusionMatrix(List<T> classes) Creates an empty confusion Matrix

Method Summary

Modifier and Type	Method and Description
void	add(ConfusionMatrix<T> other) Adds the entries from another confusion matrix to this one.
void	add(T actual, T predicted) Increments the entry specified by actual and predicted by one.
void	add(T actual, T predicted, int count) Increments the entry specified by actual and predicted by count.
boolean	equals(Object o)
int	getActualTotal(T actual) Computes the total number of times the class actually appeared in the data.
List<T>	getClasses() Gives the applyTransformToDestination of all classes in the confusion matrix.
int	getCount(T actual, T predicted) Gives the count of the number of times the "predicted" class was predicted for the "actual" class.
int	getPredictedTotal(T predicted) Computes the total number of times the class was predicted by the classifier.
int	hashCode()
String	toCSV() Outputs the ConfusionMatrix as comma-separated values for easy import into spreadsheets
String	toHTML() Outputs Confusion Matrix in an HTML table.
String	toString()

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Constructor Detail

ConfusionMatrix

public ConfusionMatrix(List<T> classes)

Creates an empty confusion Matrix

ConfusionMatrix

public ConfusionMatrix()

ConfusionMatrix

public ConfusionMatrix(ConfusionMatrix<T> other)

Creates a new ConfusionMatrix initialized with the contents of another ConfusionMatrix.

Method Detail

add

public void add(T actual,
                T predicted)

Increments the entry specified by actual and predicted by one.

add

public void add(T actual,
                T predicted,
                int count)

Increments the entry specified by actual and predicted by count.

add

public void add(ConfusionMatrix<T> other)

Adds the entries from another confusion matrix to this one.

getClasses

public List<T> getClasses()

Gives the applyTransformToDestination of all classes in the confusion matrix.

getCount

public int getCount(T actual,
                    T predicted)

Gives the count of the number of times the "predicted" class was predicted for the "actual" class.

getPredictedTotal

public int getPredictedTotal(T predicted)

Computes the total number of times the class was predicted by the classifier.

getActualTotal

public int getActualTotal(T actual)

Computes the total number of times the class actually appeared in the data.

toString

public String toString()

Overrides:

toString in class Object

toCSV

public String toCSV()

Outputs the ConfusionMatrix as comma-separated values for easy import into spreadsheets

toHTML

public String toHTML()

Outputs Confusion Matrix in an HTML table. Cascading Style Sheets (CSS) can control the table's appearance by defining the empty-space, actual-count-header, predicted-class-header, and count-element classes. For example

Returns:

html string

equals

public boolean equals(Object o)

Overrides:

equals in class Object

hashCode

public int hashCode()

Overrides:

hashCode in class Object