src/HOL/Tools/Sledgehammer/MaSh/src/sparseNaiveBayes.py
changeset 54149 70456a8f5e6e
parent 53957 ce12e547e6bb
child 54432 68f8bd1641da
--- a/src/HOL/Tools/Sledgehammer/MaSh/src/sparseNaiveBayes.py	Fri Oct 18 10:43:21 2013 +0200
+++ b/src/HOL/Tools/Sledgehammer/MaSh/src/sparseNaiveBayes.py	Fri Oct 18 13:30:09 2013 +0200
@@ -78,6 +78,8 @@
             self.counts[dep][0] -= 1
             for f,_w in features.items():
                 self.counts[dep][1][f] -= 1
+                if self.counts[dep][1][f] == 0:
+                    del self.counts[dep][1][f]
 
 
     def overwrite(self,problemId,newDependencies,dicts):