src/HOL/Tools/Sledgehammer/MaSh/src/sparseNaiveBayes.py
changeset 54692 5ce1b9613705
parent 54432 68f8bd1641da
child 54697 b08e1bbde10a
--- a/src/HOL/Tools/Sledgehammer/MaSh/src/sparseNaiveBayes.py	Sat Dec 07 18:06:49 2013 +0100
+++ b/src/HOL/Tools/Sledgehammer/MaSh/src/sparseNaiveBayes.py	Mon Dec 09 04:03:30 2013 +0100
@@ -112,7 +112,19 @@
             for f,w in features.iteritems():
                 # DEBUG
                 #w = 1.0
-                if f in fA:
+                # Test for multiple features
+                isMatch = False
+                if not isinstance( f, ( int, long ) ):
+                    f = f[0]
+                    inter = set(f).intersection(fA)
+                    if len(inter) > 0:
+                        isMatch = True
+                else:
+                    if f in fA:
+                        isMatch = True
+                 
+                if isMatch:
+                #if f in fA:
                     if fWeightsA[f] == 0:
                         resultA += w*self.defVal
                     else: