mods to Confusion Matrix

sightmachine · Feb 20, 2013 · d8610a4 · d8610a4
1 parent 82eead5
commit d8610a4
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/SimpleCV/MachineLearning/ConfusionMatrix.py b/SimpleCV/MachineLearning/ConfusionMatrix.py
@@ -25,10 +25,16 @@ def addDataPoint(self,truth_name,test_name):
         self.totalCount += 1
 
     def getCorrectPercent(self):
-        return np.around(float(self.correctCount)/float(self.totalCount),4)
+        if( self.totalCount > 0 and self.correctCount ):
+            return np.around(float(self.correctCount)/float(self.totalCount),4)
+        else:
+            return 0.00
 
     def getIncorrectPercent(self):
-        return np.around(float(self.incorrectCount)/float(self.totalCount),4)
+        if( self.totalCount > 0 and self.correctCount ):
+            return np.around(float(self.incorrectCount)/float(self.totalCount),4)
+        else:
+            return 0.00
 
     def getClassCorrectPercent(self, className):
         total = float(np.sum(self.confusionMatrix[:,self.nameMap[className]]))