Work from Google Summer of Code, 2018

.gitignore

@@ -3,8 +3,6 @@ __pycache__/
 *.py[cod]
 *$py.class
 
-# C extensions
-*.so
 
 # Distribution / packaging
 .Python

python2/pracmln/mln/inference/.gitignore

@@ -1 +1,6 @@
-*.pyo
+*.pyo
+*.c
+*.html
+setup.py
+build/*
+pracmln/*

python2/pracmln/mln/inference/exact.py → python2/pracmln/mln/inference/exact.pyx

@@ -36,6 +36,7 @@
 from numpy.ma.core import exp
 from pracmln.mln.inference.infer import Inference
 from pracmln.logic.common import Logic
+from numpy import zeros
 
 
 logger = logs.getlogger(__name__)
@@ -46,12 +47,12 @@
 
 
 def eval_queries(world):
-    """
+    '''
     Evaluates the queries given a possible world.
-    """
-    numerators = [0] * len(global_enumAsk.queries)
+    '''
+    numerators = zeros(len(global_enumAsk.queries))
     denominator = 0
-    expsum = 0
+    cdef expsum = 0
     for gf in global_enumAsk.grounder.itergroundings():
         if global_enumAsk.soft_evidence_formula(gf):
             expsum += gf.noisyor(world) * gf.weight
@@ -67,6 +68,7 @@ def eval_queries(world):
             expsum += gf(world) * gf.weight
     expsum = exp(expsum)
     # update numerators
+    cdef int i
     for i, query in enumerate(global_enumAsk.queries):
         if query(world):
             numerators[i] += expsum
@@ -115,13 +117,13 @@ def _run(self):
         for variable in self.mrf.variables:
             values = variable.valuecount(self.mrf.evidence)
             worlds *= values
-        numerators = [0.0 for i in range(len(self.queries))]
-        denominator = 0.
+        numerators = zeros(len(self.queries))#[0.0 for i in range(len(self.queries))]
+        cdef double denominator = 0.
         # start summing
         logger.debug("Summing over %d possible worlds..." % worlds)
         if worlds > 500000 and self.verbose:
             print colorize('!!! %d WORLDS WILL BE ENUMERATED !!!' % worlds, (None, 'red', True), True)
-        k = 0
+        cdef int k = 0
         self._watch.tag('enumerating worlds', verbose=self.verbose)
         global global_enumAsk
         global_enumAsk = self
@@ -135,9 +137,11 @@ def _run(self):
                 for num, denum in pool.imap(with_tracing(eval_queries), self.mrf.worlds()):
                     denominator += denum
                     k += 1
-                    for i, v in enumerate(num):
-                        numerators[i] += v
-                    if self.verbose: bar.inc()
+                    numerators += num
+                    #for i, v in enumerate(num):
+                    #    numerators[i] += v
+                    if self.verbose:
+                        bar.inc()
             except Exception as e:
                 logger.error('Error in child process. Terminating pool...')
                 pool.close()
@@ -150,8 +154,9 @@ def _run(self):
                 # compute exp. sum of weights for this world
                 num, denom = eval_queries(world)
                 denominator += denom
-                for i, _ in enumerate(self.queries):
-                    numerators[i] += num[i]
+                numerators += num
+                #for i, _ in enumerate(self.queries):
+                #    numerators[i] += num[i]
                 k += 1
                 if self.verbose:
                     bar.update(float(k) / worlds)
@@ -163,7 +168,8 @@ def _run(self):
             raise SatisfiabilityException(
                 'MLN is unsatisfiable. All probability masses returned 0.')
         # normalize answers
-        dist = map(lambda x: float(x) / denominator, numerators)
+        dist = numerators / denominator
+        #dist = map(lambda x: float(x) / denominator, numerators)
         result = {}
         for q, p in zip(self.queries, dist):
             result[str(q)] = p

python3/build-cython.sh

@@ -0,0 +1,21 @@
+
+cd pracmln/logic/
+echo "======================LOGIC======================"
+python3 setup.py build_ext --inplace
+echo "================================================="
+
+echo "=======================MLN======================="
+cd ../mln/
+python3 setup.py build_ext --inplace
+echo "================================================="
+
+echo "====================GROUNDING===================="
+cd grounding/
+python3 setup.py build_ext --inplace
+echo "================================================="
+
+echo "====================INFERENCE===================="
+cd ../inference/
+python3 setup.py build_ext --inplace
+echo "================================================="
+

python3/pracmln/__init__.py

@@ -29,4 +29,4 @@
 from .mlnlearn import QUERY_PREDS
 from .mlnlearn import EVIDENCE_PREDS
 from .utils.project import mlnpath
-from .utils.project import PRACMLNConfig
+from .utils.project import PRACMLNConfig

python3/pracmln/logic/.gitignore

@@ -0,0 +1,6 @@
+common.py
+*.pyo
+*.c
+*.html
+build/*
+pracmln/*

python3/pracmln/logic/common.cpython-35m-x86_64-linux-gnu.so

@@ -0,0 +1 @@
+pracmln/logic/common.cpython-35m-x86_64-linux-gnu.so

python3/pracmln/logic/common.pxd

@@ -0,0 +1,88 @@
+from ..mln.base cimport MLN
+
+
+
+cdef class Constraint():
+    cdef dict __dict__
+    cpdef gndatoms(self, l=*)
+
+cdef class Formula(Constraint):
+    cdef MLN _mln
+    #cdef public list children
+    pass
+
+cdef class ComplexFormula(Formula):
+    pass
+
+cdef class Conjunction(ComplexFormula):
+    cpdef int maxtruth(self, world)
+    cpdef int mintruth(self, world)
+
+cdef class Disjunction(ComplexFormula):
+    cpdef int maxtruth(self, world)
+    cpdef int mintruth(self, world)
+
+cdef class Lit(Formula):
+    cdef int _negated
+    cdef str _predname
+
+cdef class LitGroup(Formula):
+    cdef int _negated
+    cdef str _predname
+
+cdef class GroundLit(Formula):
+    cdef GroundAtom _gndatom
+    cdef int _negated
+    cpdef truth(self, list world)
+    cpdef mintruth(self, list world)
+    cpdef maxtruth(self, list world)
+
+cdef class GroundAtom():
+    cdef str _predname
+    cdef MLN mln
+    #cdef int _idx
+    cdef dict __dict__
+    cpdef truth(self, list world)
+    cpdef mintruth(self, list world)
+    cpdef maxtruth(self, list world)
+
+cdef class Equality(ComplexFormula):
+    cdef int _negated
+    cdef str _argsA
+    cdef str _argsB
+    cpdef truth(self, world=*)
+    cpdef int maxtruth(self, world)
+    cpdef int mintruth(self, world)
+
+cdef class Implication(ComplexFormula):
+    pass
+
+cdef class Biimplication(ComplexFormula):
+    pass
+
+cdef class Negation(ComplexFormula):
+    cpdef truth(self, list world)
+    pass
+
+cdef class Exist(ComplexFormula):
+    pass
+
+cdef class TrueFalse(Formula):
+    cdef float _value
+    cpdef float truth(self, world=*)
+    cpdef mintruth(self, world=*)
+    cpdef maxtruth(self, world=*)
+
+cdef class NonLogicalConstraint(Constraint):
+    pass
+
+cdef class CountConstraint(NonLogicalConstraint):
+    pass
+
+cdef class GroundCountConstraint(NonLogicalConstraint):
+    pass
+
+cdef class Logic:
+    cdef dict __dict__
+    #cdef class Constraint():
+    #    pass