From c9ab282dfa44eebafe36a9d0e978549daba161e0 Mon Sep 17 00:00:00 2001 From: Gonzague de Carpentier Date: Sat, 10 Sep 2022 20:19:54 +0200 Subject: [PATCH] update docstring of ot.gromov.init_matrix --- ot/gromov.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/ot/gromov.py b/ot/gromov.py index 7de85988e..bc1c8e593 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 -------