From f15819e8bcac65b49982ebedc06023aafee967dc Mon Sep 17 00:00:00 2001 From: Andreas Schuh Date: Wed, 2 Aug 2023 23:32:26 +0000 Subject: [PATCH] [losses] Fix documentation of mi_loss() return value --- src/deepali/losses/functional.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/deepali/losses/functional.py b/src/deepali/losses/functional.py index d3ff967..3314541 100644 --- a/src/deepali/losses/functional.py +++ b/src/deepali/losses/functional.py @@ -1005,7 +1005,7 @@ def mi_loss( normalized: Calculate Normalized Mutual Information instead of Mutual Information if True. Returns: - Negative mutual information. If ``normalized=True``, 2 is added such that the loss value is in [0, 1]. + Negative mutual information. If ``normalized=True``, 2 is added such that the loss value is in [0, 2]. """