Update graph_semiring.py

ML-KULeuven · Jan 4, 2024 · 04cf646 · 04cf646
1 parent c777cce
commit 04cf646
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/deepproblog/semiring/graph_semiring.py b/src/deepproblog/semiring/graph_semiring.py
@@ -113,9 +113,9 @@ def cross_entropy(
             )
         else:
             if target == 1.0:
-                loss = -torch.log(p) * weight
+                loss = -torch.log(p + eps) * weight
             elif target == 0.0:
-                loss = -torch.log(1.0 - p) * weight
+                loss = -torch.log(1.0 - p + eps) * weight
             else:
                 loss = (
                     -(