Merge branch 'dev-vs' of github.com:VincentStimper/normalizing-flows …

…into dev-vs
dataflowr · Jan 23, 2023 · 75609e1 · 75609e1
2 parents 70cfa36 + b3a9a45
commit 75609e1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/normflows/distributions/prior.py b/normflows/distributions/prior.py
@@ -108,8 +108,13 @@ class TwoModes(PriorDistribution):
     def __init__(self, loc, scale):
         """Distribution 2d with two modes
 
-        Distribution 2d with two modes at:
+        Distribution 2d with two modes at
         ```z[0] = -loc```  and ```z[0] = loc```
+        following the density
+        ```
+        log(p) = 1/2 * ((norm(z) - loc) / (2 * scale)) ** 2
+                - log(exp(-1/2 * ((z[0] - loc) / (3 * scale)) ** 2) + exp(-1/2 * ((z[0] + loc) / (3 * scale)) ** 2))
+        ```
 
         Args:
           loc: distance of modes from the origin