diff options
author | Hicham Janati <hicham.janati@inria.fr> | 2019-08-28 15:40:40 +0200 |
---|---|---|
committer | Hicham Janati <hicham.janati@inria.fr> | 2019-08-28 15:40:40 +0200 |
commit | 7f4af32c70452ff12fa8c15dea317c14d3d97878 (patch) | |
tree | 22bfe624ac01cafef9e48f308f3d8f4d6842efee | |
parent | cfdbbd21642c6082164b84db78c2ead07499a113 (diff) |
correct call of unbalanced with reg_m
-rw-r--r-- | ot/da.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -1902,7 +1902,7 @@ class UnbalancedSinkhornTransport(BaseTransport): returned_ = sinkhorn_unbalanced( a=self.mu_s, b=self.mu_t, M=self.cost_, - reg=self.reg_e, alpha=self.reg_m, method=self.method, + reg=self.reg_e, reg_m=self.reg_m, method=self.method, numItermax=self.max_iter, stopThr=self.tol, verbose=self.verbose, log=self.log) |