From b295ffccc95149c1d63e805e1ca6f027a4071e2a Mon Sep 17 00:00:00 2001 From: decarpentierg <82534773+decarpentierg@users.noreply.github.com> Date: Mon, 12 Sep 2022 08:48:16 +0200 Subject: update docstring of ot.gromov.init_matrix (#398) Co-authored-by: Gonzague de Carpentier --- ot/gromov.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/ot/gromov.py b/ot/gromov.py index 7de8598..bc1c8e5 100644 --- a/ot/gromov.py +++ b/ot/gromov.py @@ -70,9 +70,12 @@ def init_matrix(C1, C2, p, q, loss_fun='square_loss'): Metric cost matrix in the source space C2 : array-like, shape (nt, nt) Metric cost matrix in the target space - T : array-like, shape (ns, nt) - Coupling between source and target spaces p : array-like, shape (ns,) + Probability distribution in the source space + q : array-like, shape (nt,) + Probability distribution in the target space + loss_fun : str, optional + Name of loss function to use: either 'square_loss' or 'kl_loss' (default='square_loss') Returns ------- -- cgit v1.2.3