diff options
| author | Rémi Flamary <remi.flamary@gmail.com> | 2018-09-24 10:17:21 +0200 |
|---|---|---|
| committer | Rémi Flamary <remi.flamary@gmail.com> | 2018-09-24 10:17:21 +0200 |
| commit | 24a53ef2dba0a43c282f6b31937c3e7901df7930 (patch) | |
| tree | b10800380f0d3b376489e717ce23f29a09916572 /ot | |
| parent | 7ffd4fef3260e086b0b1ed050f5cb4b83195b122 (diff) | |
add contributor
Diffstat (limited to 'ot')
| -rw-r--r-- | ot/bregman.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index faa6365..97027e8 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -536,7 +536,7 @@ def greenkhorn(a, b, M, reg, numItermax=10000, stopThr=1e-9, verbose=False, log= G[i_1, :] = u[i_1] * K[i_1, :] * v viol[i_1] = u[i_1] * K[i_1, :].dot(v) - a[i_1] - viol_2 = viol_2 + (K[i_1, :].T * (u[i_1] - old_u) * v) + viol_2 += (K[i_1, :].T * (u[i_1] - old_u) * v) else: old_v = v[i_2] @@ -544,7 +544,7 @@ def greenkhorn(a, b, M, reg, numItermax=10000, stopThr=1e-9, verbose=False, log= G[:, i_2] = u * K[:, i_2] * v[i_2] #aviol = (G@one_m - a) #aviol_2 = (G.T@one_n - b) - viol = viol + (-old_v + v[i_2]) * K[:, i_2] * u + viol += (-old_v + v[i_2]) * K[:, i_2] * u viol_2[i_2] = v[i_2] * K[:, i_2].dot(u) - b[i_2] #print('b',np.max(abs(aviol -viol)),np.max(abs(aviol_2 - viol_2))) |