From 73a0122eadd84ed5adc89de347b3ec5e316f5fa8 Mon Sep 17 00:00:00 2001 From: Tyson Bailey Date: Wed, 26 Aug 2015 22:44:46 -0600 Subject: [PATCH] Inserted check for confusion. --- SimpleCV/MachineLearning/NaiveBayesClassifier.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/SimpleCV/MachineLearning/NaiveBayesClassifier.py b/SimpleCV/MachineLearning/NaiveBayesClassifier.py index ae821d553..065d1d672 100644 --- a/SimpleCV/MachineLearning/NaiveBayesClassifier.py +++ b/SimpleCV/MachineLearning/NaiveBayesClassifier.py @@ -237,8 +237,9 @@ def train(self,images,classNames,disp=None,subset=-1,savedata=None,verbose=True) print("Incorrect: "+str(bad)) classes = self.mDataSetOrange.domain.classVar.values print "\t"+"\t".join(classes) - for className, classConfusions in zip(classes, confusion): - print ("%s" + ("\t%i" * len(classes))) % ((className, ) + tuple(classConfusions)) + if confusion > 0: + for className, classConfusions in zip(classes, confusion): + print ("%s" + ("\t%i" * len(classes))) % ((className, ) + tuple(classConfusions)) return [good, bad, confusion]