sigma estimate fix

ondrolexa · Oct 6, 2023 · b1f9afd · b1f9afd
1 parent fa301ff
commit b1f9afd
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/apsg/plotting/_stereogrid.py b/src/apsg/plotting/_stereogrid.py
@@ -108,10 +108,11 @@ def calculate_density(self, features, **kwargs):
             # k = estimate_k(features)
             # sigma = np.sqrt(2 * n / (k - 2))
             # Totally empirical as estimate_k is problematic
-            sigma = np.sqrt(2 * n / (np.log(n) - 2)) / 3
-            k = 2 * (1.0 + n / sigma**2)
-        else:
-            k = 2 * (1.0 + n / sigma**2)
+            if n < 10:
+                sigma = 3
+            else:
+                sigma = np.sqrt(2 * n / (np.log(n) - 2)) / 3
+        k = 2 * (1.0 + n / sigma**2)
         # method = kwargs.get("method", "exp_kamb")
         trim = kwargs.get("trimzero", True)
         sigmanorm = kwargs.get("sigmanorm", True)