From 6ededec52c3135ac25068ade51b6c196e153679b Mon Sep 17 00:00:00 2001
From: Kazuki Adachi <kazuki.adachi.xy@gmail.com>
Date: Thu, 4 Apr 2024 21:06:14 +0900
Subject: [PATCH] modify formula of docstring

---
 ignite/metrics/mutual_information.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/ignite/metrics/mutual_information.py b/ignite/metrics/mutual_information.py
index b01c967dce1..50e7d9b1b74 100644
--- a/ignite/metrics/mutual_information.py
+++ b/ignite/metrics/mutual_information.py
@@ -15,11 +15,10 @@ class MutualInformation(Metric):
 
     .. math::
 
-       I(X;Y) = H(Y) - H(Y|X)
+       I(X;Y) &= H(Y) - H(Y|X)
        = H \left( \frac{1}{N}\sum_{i=1}^N \hat{\mathbf{p}}_i \right)
-       - \frac{1}{N}\sum_{i=1}^N H(\hat{\mathbf{p}}_i),
-
-       H(\mathbf{p}) = -\sum_{c=1}^C p_c \log p_c.
+       - \frac{1}{N}\sum_{i=1}^N H(\hat{\mathbf{p}}_i), \\
+       H(\mathbf{p}) &= -\sum_{c=1}^C p_c \log p_c.
 
     where :math:`\hat{\mathbf{p}}_i` is the prediction probability vector for :math:`i`-th input,
     and :math:`H(\mathbf{p})` is the entropy of :math:`\mathbf{p}`.