Skip to content

Commit

Permalink
swap the definitions of p and q
Browse files Browse the repository at this point in the history
  • Loading branch information
kzkadc committed Apr 17, 2024
1 parent 273dece commit 414b25a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion ignite/metrics/js_divergence.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ class JSDivergence(KLDivergence):
D_\text{KL}(\mathbf{p}_i \| \mathbf{q}_i) &= \sum_{c=1}^C p_{i,c} \log \frac{p_{i,c}}{q_{i,c}}.
\end{align*}
where :math:`\mathbf{p}_i` and :math:`\mathbf{q}_i` are the prediction and ground truth probability tensors,
where :math:`\mathbf{p}_i` and :math:`\mathbf{q}_i` are the ground truth and prediction probability tensors,
and :math:`D_\text{KL}` is the KL-divergence.
- ``update`` must receive output of the form ``(y_pred, y)``.
Expand Down

0 comments on commit 414b25a

Please sign in to comment.