diff options
| author | Kilian Fatras <kilianfatras@dhcp-206-12-53-92.eduroam.wireless.ubc.ca> | 2018-06-21 17:52:05 -0700 |
|---|---|---|
| committer | Kilian Fatras <kilianfatras@dhcp-206-12-53-92.eduroam.wireless.ubc.ca> | 2018-06-21 17:52:05 -0700 |
| commit | e8cf3cc343c934b7c49d303186cdf226204813b3 (patch) | |
| tree | 310515612da8cb028eb28810050cb853333eae88 /ot/stochastic.py | |
| parent | af5f726f6adb52457ca6730ffb85f2ab486b2ada (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 374d1a5..f4d4c7d 100644 --- a/ot/stochastic.py +++ b/ot/stochastic.py @@ -153,7 +153,7 @@ def sag_entropic_transport(a, b, M, reg, numItermax=10000, lr=None): ''' if lr is None: - lr = 1. / max(a/reg) + lr = 1. / max(a / reg) n_source = np.shape(M)[0] n_target = np.shape(M)[1] cur_beta = np.zeros(n_target) @@ -238,7 +238,7 @@ def averaged_sgd_entropic_transport(a, b, M, reg, numItermax=300000, lr=None): ''' if lr is None: - lr = 1. / max(a/reg) + lr = 1. / max(a / reg) n_source = np.shape(M)[0] n_target = np.shape(M)[1] cur_beta = np.zeros(n_target) |