diff options
author | tgnassou <66993815+tgnassou@users.noreply.github.com> | 2023-02-28 08:35:12 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-02-28 08:35:12 +0100 |
commit | a6d5d75c6ca584ab9736b528810b3595f2571d82 (patch) | |
tree | 1fb295f86cc055222cc223ae00d05a674bb3a46c /ot | |
parent | a313e21f223af16cf21d3b7dd01bd0c6345d574c (diff) |
[MRG] Add method argument to sinkhorn Transport (#440)
* add method argument to sinkhron transport'
* update release
Diffstat (limited to 'ot')
-rw-r--r-- | ot/da.py | 5 |
1 files changed, 3 insertions, 2 deletions
@@ -1417,12 +1417,13 @@ class SinkhornTransport(BaseTransport): Sciences, 7(3), 1853-1882. """ - def __init__(self, reg_e=1., max_iter=1000, + def __init__(self, reg_e=1., method="sinkhorn", max_iter=1000, tol=10e-9, verbose=False, log=False, metric="sqeuclidean", norm=None, distribution_estimation=distribution_estimation_uniform, out_of_sample_map='ferradans', limit_max=np.infty): self.reg_e = reg_e + self.method = method self.max_iter = max_iter self.tol = tol self.verbose = verbose @@ -1463,7 +1464,7 @@ class SinkhornTransport(BaseTransport): # coupling estimation returned_ = sinkhorn( a=self.mu_s, b=self.mu_t, M=self.cost_, reg=self.reg_e, - numItermax=self.max_iter, stopThr=self.tol, + method=self.method, numItermax=self.max_iter, stopThr=self.tol, verbose=self.verbose, log=self.log) # deal with the value of log |