diff options
-rw-r--r-- | ot/bregman.py | 4 | ||||
-rw-r--r-- | ot/gpu/bregman.py | 2 | ||||
-rw-r--r-- | ot/gpu/da.py | 3 |
3 files changed, 4 insertions, 5 deletions
diff --git a/ot/bregman.py b/ot/bregman.py index 929388e..d63c51d 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -110,7 +110,7 @@ def sinkhorn(a, b, M, reg, method='sinkhorn', numItermax=1000, stopThr=1e-9, ver def sink(): return sinkhorn_epsilon_scaling( a, b, M, reg, numItermax=numItermax, - stopThr=stopThr, verbose=verbose, log=log, **kwargs) + stopThr=stopThr, verbose=verbose, log=log, **kwargs) else: print('Warning : unknown method using classic Sinkhorn Knopp') @@ -219,7 +219,7 @@ def sinkhorn2(a, b, M, reg, method='sinkhorn', numItermax=1000, stopThr=1e-9, ve def sink(): return sinkhorn_epsilon_scaling( a, b, M, reg, numItermax=numItermax, - stopThr=stopThr, verbose=verbose, log=log, **kwargs) + stopThr=stopThr, verbose=verbose, log=log, **kwargs) else: print('Warning : unknown method using classic Sinkhorn Knopp') diff --git a/ot/gpu/bregman.py b/ot/gpu/bregman.py index 86bfec1..47939c4 100644 --- a/ot/gpu/bregman.py +++ b/ot/gpu/bregman.py @@ -4,7 +4,7 @@ Bregman projections for regularized OT with GPU """ # Author: Remi Flamary <remi.flamary@unice.fr> -# Leo Gautheron <https://github.com/aje> +# Leo Gautheron <https://github.com/aje> # # License: MIT License diff --git a/ot/gpu/da.py b/ot/gpu/da.py index 7fb488d..05c580f 100644 --- a/ot/gpu/da.py +++ b/ot/gpu/da.py @@ -6,12 +6,11 @@ Domain adaptation with optimal transport with GPU implementation # Author: Remi Flamary <remi.flamary@unice.fr> # Nicolas Courty <ncourty@irisa.fr> # Michael Perrot <michael.perrot@univ-st-etienne.fr> -# Leo Gautheron <https://github.com/aje> +# Leo Gautheron <https://github.com/aje> # # License: MIT License - import numpy as np from ..utils import unif from ..da import OTDA |