summaryrefslogtreecommitdiff
path: root/ot/stochastic.py
diff options
context:
space:
mode:
authorKilian Fatras <kilianfatras@dhcp-206-12-53-99.eduroam.wireless.ubc.ca>2018-06-15 19:18:20 -0700
committerKilian Fatras <kilianfatras@dhcp-206-12-53-99.eduroam.wireless.ubc.ca>2018-06-15 19:18:20 -0700
commit3617b42ec05fe88eb616b32b2c3827d63022e287 (patch)
tree31305cd1062880c25d92591bccf8096e65896cb8 /ot/stochastic.py
parent18f9242b9575e7551d5e30b66cfbaaac85e0022a (diff)
PEP8
Diffstat (limited to 'ot/stochastic.py')
-rw-r--r--ot/stochastic.py3
1 files changed, 0 insertions, 3 deletions
diff --git a/ot/stochastic.py b/ot/stochastic.py
index 541f456..f34154b 100644
--- a/ot/stochastic.py
+++ b/ot/stochastic.py
@@ -313,7 +313,6 @@ def c_transform_entropic(b, M, reg, v):
'''
n_source = np.shape(M)[0]
- n_target = np.shape(M)[1]
u = np.zeros(n_source)
for i in range(n_source):
r = M[i, :] - v
@@ -399,8 +398,6 @@ def transportation_matrix_entropic(a, b, M, reg, method, numItermax=10000,
Advances in Neural Information Processing Systems (2016),
arXiv preprint arxiv:1605.08527.
'''
- n_source = 7
- n_target = 4
if method.lower() == "sag":
opt_v = sag_entropic_transport(a, b, M, reg, numItermax, lr)
elif method.lower() == "asgd":