From e70b9ceba942b9221e31c68894ed16705feaf3bd Mon Sep 17 00:00:00 2001 From: Daniel Nagel Date: Mon, 11 Mar 2024 15:38:59 +0100 Subject: [PATCH] Update _estimators.py --- src/normi/_estimators.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/normi/_estimators.py b/src/normi/_estimators.py index 92f2448..1cffac0 100644 --- 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