diff options
| author | Kilian Fatras <kilianfatras@dhcp-206-12-53-99.eduroam.wireless.ubc.ca> | 2018-06-15 19:25:29 -0700 |
|---|---|---|
| committer | Kilian Fatras <kilianfatras@dhcp-206-12-53-99.eduroam.wireless.ubc.ca> | 2018-06-15 19:25:29 -0700 |
| commit | 055417ee06917ff8bac5d07b2d2a17d80e5da4b6 (patch) | |
| tree | 0306fdcb471b6172f85a9e545af67f0c548bec56 /ot/stochastic.py | |
| parent | 3617b42ec05fe88eb616b32b2c3827d63022e287 (diff) | |
pep8
Diffstat (limited to 'ot/stochastic.py')
| -rw-r--r-- | ot/stochastic.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ot/stochastic.py b/ot/stochastic.py index f34154b..9912223 100644 --- a/ot/stochastic.py +++ b/ot/stochastic.py @@ -407,6 +407,6 @@ def transportation_matrix_entropic(a, b, M, reg, method, numItermax=10000, return None opt_u = c_transform_entropic(b, M, reg, opt_v) - pi = (np.exp((opt_u[:, None] + opt_v[None, :] - M[:, :]) / reg) - * a[:, None] * b[None, :]) + pi = (np.exp((opt_u[:, None] + opt_v[None, :] - M[:, :]) / reg) * + a[:, None] * b[None, :]) return pi |