Add Mode to Rice (#649)

arviz-devs · Feb 21, 2025 · 4aad654 · 4aad654
1 parent f21909f
commit 4aad654
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/preliz/distributions/rice.py b/preliz/distributions/rice.py
@@ -3,7 +3,7 @@
 
 from preliz.distributions.distributions import Continuous
 from preliz.internal.distribution_helper import all_not_none, eps
-from preliz.internal.optimization import optimize_ml, optimize_moments_rice
+from preliz.internal.optimization import find_mode, optimize_ml, optimize_moments_rice
 from preliz.internal.special import cdf_bounds, ppf_bounds_cont
 
 
@@ -129,6 +129,9 @@ def entropy(self):
     def mean(self):
         return self.sigma * np.sqrt(np.pi / 2) * _l_half(-(self.nu**2) / (2 * self.sigma**2))
 
+    def mode(self):
+        return find_mode(self)
+
     def median(self):
         return self.ppf(0.5)