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]]))