Skip to content

Commit f76f4aa

Browse files
committed
take care of log1p as well
1 parent 88c72ee commit f76f4aa

File tree

1 file changed

+4
-1
lines changed
  • bayesflow/adapters/transforms

1 file changed

+4
-1
lines changed

bayesflow/adapters/transforms/log.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,10 @@ def get_config(self) -> dict:
3939
return serialize({"p1": self.p1})
4040

4141
def log_det_jac(self, data: np.ndarray, inverse: bool = False, **kwargs) -> np.ndarray:
42-
ldj = -np.log(data)
42+
if self.p1:
43+
ldj = -np.log1p(data)
44+
else:
45+
ldj = -np.log(data)
4346
if inverse:
4447
ldj = -ldj
4548
return np.sum(ldj, axis=tuple(range(1, ldj.ndim)))

0 commit comments

Comments
 (0)