From 2f3741299989ffb105bed986f7a85d567fa6cb6a Mon Sep 17 00:00:00 2001 From: Hicham Janati Date: Sat, 19 Oct 2019 12:05:45 +0200 Subject: add weights arg in generic barycenter func --- ot/bregman.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'ot/bregman.py') diff --git a/ot/bregman.py b/ot/bregman.py index 2cd832b..ba5c7ba 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -1037,11 +1037,13 @@ def barycenter(A, M, reg, weights=None, method="sinkhorn", numItermax=10000, """ if method.lower() == 'sinkhorn': - return barycenter_sinkhorn(A, M, reg, numItermax=numItermax, + return barycenter_sinkhorn(A, M, reg, weights=weights, + numItermax=numItermax, stopThr=stopThr, verbose=verbose, log=log, **kwargs) elif method.lower() == 'sinkhorn_stabilized': - return barycenter_stabilized(A, M, reg, numItermax=numItermax, + return barycenter_stabilized(A, M, reg, weights=weights, + numItermax=numItermax, stopThr=stopThr, verbose=verbose, log=log, **kwargs) else: -- cgit v1.2.3