set log-potentials to -infinity instead of zero

remilepriol · Jan 3, 2017 · bbf2f74 · bbf2f74
1 parent 120f951
commit bbf2f74
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/IsingModel/Ising2.py b/IsingModel/Ising2.py
@@ -172,10 +172,12 @@ def loopybelief(self, max_iter=25):
         # 4 : log-potentials in each point
         # potentials are stored in an additional channel
         messages[4] = self.__repeat_symmetric(self.linear_factors)
-        # if +1 is observed, set potential for -1 to 0
-        messages[4, :, :, 1] = (1-(self.observations == 1)) * messages[4, :, :, 1]
-        # if -1 is observed, set potential for +1 to 0
-        messages[4, :,:, 0] = (1-(self.observations == -1)) * messages[4, :, :, 1]
+        # if +1 is observed, set potential for -1 to -infinity
+        messages[4, :, :, 1] = (1 - (self.observations == 1)) * messages[4, :, :, 1] + \
+            float('-inf') * (self.observations == 1)
+        # if -1 is observed, set potential for +1 to -infinity
+        messages[4, :, :, 0] = (1 - (self.observations == -1)) * messages[4, :, :, 0] + \
+            float('-inf') * (self.observations == -1)
 
         upper = messages[:, :-1, :]
         lower = messages[:, 1:, :]