diff options
| -rw-r--r-- | ot/gromov.py | 7 |
1 files 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
-------
|