Merge pull request #5 from moldyn/Fix-numerical-issue-for-small-numbe…

…r-of-samples Fix beartype warning for almost I=0 but negative
moldyn · Mar 11, 2024 · eff9c20 · eff9c20
2 parents 28b8216 + e70b9ce
commit eff9c20
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/normi/_estimators.py b/src/normi/_estimators.py
@@ -428,7 +428,7 @@ def kraskov_estimator(
     mean_log_eps: Float = np.mean(np.log(radii))
 
     return (
-        digamma_N + digamma_k - digamma_nx - digamma_ny,  # mi
+        np.max([digamma_N + digamma_k - digamma_nx - digamma_ny, 0]),  # mi
         digamma_N - digamma_k + (dx + dy) * mean_log_eps,  # hxy
         digamma_N - digamma_nx + dx * mean_log_eps,  # hx
         digamma_N - digamma_ny + dy * mean_log_eps,  # hy