From d8610a4dfbc9099151343dcb7fe355c84874d69a Mon Sep 17 00:00:00 2001 From: kscottz Date: Wed, 20 Feb 2013 12:40:38 -0500 Subject: [PATCH] mods to Confusion Matrix --- SimpleCV/MachineLearning/ConfusionMatrix.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/SimpleCV/MachineLearning/ConfusionMatrix.py b/SimpleCV/MachineLearning/ConfusionMatrix.py index e92235c8d..321cf2bad 100644 --- 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]]))