diff options
Diffstat (limited to 'ot/bregman.py')
-rw-r--r-- | ot/bregman.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index ad9a67a..2d82ae4 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -6,12 +6,12 @@ Bregman projections for regularized OT import numpy as np -def sinkhorn(a,b, M, reg,numItermax = 1000,stopThr=1e-9,verbose=False,log=False): +def sinkhorn(a,b, M, reg, numItermax = 1000, stopThr=1e-9, verbose=False, log=False): """ - Solve the entropic regularization optimal transport problem and return the OT matrix - + Solve the entropic regularization optimal transport problem + The function solves the following optimization problem: - + .. math:: \gamma = arg\min_\gamma <\gamma,M>_F + reg\cdot\Omega(\gamma) |