summaryrefslogtreecommitdiff
path: root/ot/da.py
diff options
context:
space:
mode:
authorHicham Janati <hicham.janati@inria.fr>2019-08-28 15:40:40 +0200
committerHicham Janati <hicham.janati@inria.fr>2019-08-28 15:40:40 +0200
commit7f4af32c70452ff12fa8c15dea317c14d3d97878 (patch)
tree22bfe624ac01cafef9e48f308f3d8f4d6842efee /ot/da.py
parentcfdbbd21642c6082164b84db78c2ead07499a113 (diff)
correct call of unbalanced with reg_m
Diffstat (limited to 'ot/da.py')
-rw-r--r--ot/da.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/da.py b/ot/da.py
index 2af855d..108a38d 100644
--- a/ot/da.py
+++ b/ot/da.py
@@ -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)