update

PaddlePaddle · Oct 8, 2023 · 4c1bfd4 · 4c1bfd4
1 parent 91f9e01
commit 4c1bfd4
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/python/paddle/distribution/binomial.py b/python/paddle/distribution/binomial.py
@@ -200,8 +200,7 @@ def entropy(self):
             numpy.ndarray: the entropy for the binomial r.v.
         """
         values = self._enumerate_support()
-        eps = paddle.finfo(self.probability.dtype).eps
-        log_prob = paddle.nan_to_num(self.log_prob(values), neginf=eps)
+        log_prob = self.log_prob(values)
         return -(paddle.exp(log_prob) * log_prob).sum(0)
 
     def _enumerate_support(self):
@@ -233,11 +232,16 @@ def log_prob(self, value):
             - paddle.lgamma(self.total_count - value + 1.0)
             - paddle.lgamma(value + 1.0)
         )
+        eps = paddle.finfo(self.probability.dtype).eps
+        probs = paddle.clip(self.probability, min=eps, max=1 - eps)
         # log_p
-        return (
-            log_comb
-            + value * paddle.log(self.probability)
-            + (self.total_count - value) * paddle.log(1 - self.probability)
+        return paddle.nan_to_num(
+            (
+                log_comb
+                + value * paddle.log(probs)
+                + (self.total_count - value) * paddle.log(1 - probs)
+            ),
+            neginf=-eps,
         )
 
     def prob(self, value):

diff --git a/python/paddle/distribution/poisson.py b/python/paddle/distribution/poisson.py
@@ -216,7 +216,7 @@ def log_prob(self, value):
                 + value * paddle.log(self.rate)
                 - paddle.lgamma(value + 1)
             ),
-            neginf=eps,
+            neginf=-eps,
         )
 
     def prob(self, value):