Logistic Normal added to distributions

ciemss · Oct 1, 2024 · 4f87c30 · 4f87c30
1 parent 08cf9fc
commit 4f87c30
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/pyciemss/mira_integration/distributions.py b/pyciemss/mira_integration/distributions.py
@@ -154,6 +154,19 @@ def mira_laplace_to_pyro(parameters: ParameterDict) -> pyro.distributions.Distri
 
     return pyro.distributions.Laplace(loc=loc, scale=scale)
 
+def mira_logistic_normal_to_pyro(parameters: ParameterDict) -> pyro.distributions.Distribution:
+    if "location" in parameters.keys():
+        loc = parameters["location"]
+    elif "mu" in parameters.keys():
+        loc = parameters["mu"]
+
+    if "scale" in parameters.keys():
+        scale = parameters["scale"]
+    elif "sigma" in parameters.keys():
+        scale = parameters["sigma"]
+    elif "tau" in parameters.keys():
+        scale = 1.0 / parameters["tau"]
+    return pyro.distributions.LogisticNormal(loc=loc, scale=scale)
 
 def mira_paretotypeI_to_pyro(
     parameters: ParameterDict,
@@ -226,6 +239,7 @@ def mira_weibull_to_pyro(parameters: ParameterDict) -> pyro.distributions.Distri
     "Gumbel1": mira_gumbel_to_pyro,
     "Laplace1": mira_laplace_to_pyro,
     "Laplace2": mira_laplace_to_pyro,
+    "LogitNormal1": mira_logistic_normal_to_pyro,
     "ParetoTypeI1": mira_paretotypeI_to_pyro,
     "Poisson1": mira_poisson_to_pyro,
     "StudentT1": mira_studentt_to_pyro,