org.apache.mahout.classifier
Class ConfusionMatrix

java.lang.Object
  extended by org.apache.mahout.classifier.ConfusionMatrix

public class ConfusionMatrix
extends Object

The ConfusionMatrix Class stores the result of Classification of a Test Dataset. See http://en.wikipedia.org/wiki/Confusion_matrix for background


Constructor Summary
ConfusionMatrix(Collection<String> labels, String defaultLabel)
           
 
Method Summary
 void addInstance(String correctLabel, ClassifierResult classifiedResult)
           
 void addInstance(String correctLabel, String classifiedLabel)
           
 double getAccuracy(String label)
           
 int[][] getConfusionMatrix()
           
 int getCorrect(String label)
           
 int getCount(String correctLabel, String classifiedLabel)
           
 Collection<String> getLabels()
           
 double getTotal(String label)
           
 void incrementCount(String correctLabel, String classifiedLabel)
           
 void incrementCount(String correctLabel, String classifiedLabel, int count)
           
 ConfusionMatrix merge(ConfusionMatrix b)
           
 void putCount(String correctLabel, String classifiedLabel, int count)
           
 String toString()
           
 
Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait
 

Constructor Detail

ConfusionMatrix

public ConfusionMatrix(Collection<String> labels,
                       String defaultLabel)
Method Detail

getConfusionMatrix

public int[][] getConfusionMatrix()

getLabels

public Collection<String> getLabels()

getAccuracy

public double getAccuracy(String label)

getCorrect

public int getCorrect(String label)

getTotal

public double getTotal(String label)

addInstance

public void addInstance(String correctLabel,
                        ClassifierResult classifiedResult)

addInstance

public void addInstance(String correctLabel,
                        String classifiedLabel)

getCount

public int getCount(String correctLabel,
                    String classifiedLabel)

putCount

public void putCount(String correctLabel,
                     String classifiedLabel,
                     int count)

incrementCount

public void incrementCount(String correctLabel,
                           String classifiedLabel,
                           int count)

incrementCount

public void incrementCount(String correctLabel,
                           String classifiedLabel)

merge

public ConfusionMatrix merge(ConfusionMatrix b)

toString

public String toString()
Overrides:
toString in class Object


Copyright © 2008-2011 The Apache Software Foundation. All Rights Reserved.