From 7f4af32c70452ff12fa8c15dea317c14d3d97878 Mon Sep 17 00:00:00 2001 From: Hicham Janati Date: Wed, 28 Aug 2019 15:40:40 +0200 Subject: correct call of unbalanced with reg_m --- ot/da.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ot/da.py') 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) -- cgit v1.2.3