From 439860609df786a877383775dd901afe28480cc9 Mon Sep 17 00:00:00 2001 From: ievred Date: Wed, 1 Apr 2020 09:00:03 +0200 Subject: fix imports remove checks --- ot/da.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) (limited to 'ot/da.py') diff --git a/ot/da.py b/ot/da.py index a9c3cea..e62e495 100644 --- a/ot/da.py +++ b/ot/da.py @@ -14,7 +14,7 @@ Domain adaptation with optimal transport import numpy as np import scipy.linalg as linalg -from .bregman import sinkhorn +from .bregman import sinkhorn, jcpot_barycenter from .lp import emd from .utils import unif, dist, kernel, cost_normalization from .utils import check_params, BaseEstimator @@ -1956,8 +1956,7 @@ class JCPOTTransport(BaseTransport): def __init__(self, reg_e=.1, max_iter=10, tol=10e-9, verbose=False, log=False, - metric="sqeuclidean", norm=None, - distribution_estimation=distribution_estimation_uniform, + metric="sqeuclidean", out_of_sample_map='ferradans'): self.reg_e = reg_e -- cgit v1.2.3