We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7fa8438 commit 7ee082dCopy full SHA for 7ee082d
ot/gromov/_bregman.py
@@ -373,8 +373,8 @@ def entropic_gromov_barycenters(
373
lambdas : list of float, optional
374
List of the `S` spaces' weights.
375
If let to its default value None, uniform weights are taken.
376
- loss_fun : callable, optional
377
- tensor-matrix multiplication function based on specific loss function
+ loss_fun : string, optional (default='square_loss')
+ Loss function used for the solver either 'square_loss' or 'kl_loss'
378
epsilon : float, optional
379
Regularization term >0
380
symmetric : bool, optional.
0 commit comments