Skip to content

Commit 7f4af32

Browse files
author
Hicham Janati
committed
correct call of unbalanced with reg_m
1 parent cfdbbd2 commit 7f4af32

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ot/da.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1902,7 +1902,7 @@ class label
19021902

19031903
returned_ = sinkhorn_unbalanced(
19041904
a=self.mu_s, b=self.mu_t, M=self.cost_,
1905-
reg=self.reg_e, alpha=self.reg_m, method=self.method,
1905+
reg=self.reg_e, reg_m=self.reg_m, method=self.method,
19061906
numItermax=self.max_iter, stopThr=self.tol,
19071907
verbose=self.verbose, log=self.log)
19081908

0 commit comments

Comments
 (0)