summaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
authorievred <ievgen.redko@univ-st-etienne.fr>2020-04-08 10:28:57 +0200
committerievred <ievgen.redko@univ-st-etienne.fr>2020-04-08 10:28:57 +0200
commitd6ef8676cc3f94ba5d80acc9fd9745c9ed91819a (patch)
treee9a17a904b12748ac9f7bfb602da55fe3c23d7f4 /test
parent2c9f992157844d6253a302905417e86580ac6b12 (diff)
remove jcpot from laplace
Diffstat (limited to 'test')
-rw-r--r--test/test_da.py56
1 files changed, 1 insertions, 55 deletions
diff --git a/test/test_da.py b/test/test_da.py
index 4eaf193..0e31f26 100644
--- a/test/test_da.py
+++ b/test/test_da.py
@@ -549,60 +549,6 @@ def test_linear_mapping_class():
np.testing.assert_allclose(Ct, Cst, rtol=1e-2, atol=1e-2)
-def test_jcpot_transport_class():
- """test_jcpot_transport
- """
-
- ns1 = 150
- ns2 = 150
- nt = 200
-
- Xs1, ys1 = make_data_classif('3gauss', ns1)
- Xs2, ys2 = make_data_classif('3gauss', ns2)
-
- Xt, yt = make_data_classif('3gauss2', nt)
-
- Xs = [Xs1, Xs2]
- ys = [ys1, ys2]
-
- otda = ot.da.JCPOTTransport(reg_e=0.01, max_iter=1000, tol=1e-9, verbose=True, log=True)
-
- # test its computed
- otda.fit(Xs=Xs, ys=ys, Xt=Xt)
-
- assert hasattr(otda, "coupling_")
- assert hasattr(otda, "proportions_")
- assert hasattr(otda, "log_")
-
- # test dimensions of coupling
- for i, xs in enumerate(Xs):
- assert_equal(otda.coupling_[i].shape, ((xs.shape[0], Xt.shape[0])))
-
- # test all margin constraints
- mu_t = unif(nt)
-
- for i in range(len(Xs)):
- # test margin constraints w.r.t. uniform target weights for each coupling matrix
- assert_allclose(
- np.sum(otda.coupling_[i], axis=0), mu_t, rtol=1e-3, atol=1e-3)
-
- # test margin constraints w.r.t. modified source weights for each source domain
-
- assert_allclose(
- np.dot(otda.log_['D1'][i], np.sum(otda.coupling_[i], axis=1)), otda.proportions_, rtol=1e-3,
- atol=1e-3)
-
- # test transform
- transp_Xs = otda.transform(Xs=Xs)
- [assert_equal(x.shape, y.shape) for x, y in zip(transp_Xs, Xs)]
-
- Xs_new, _ = make_data_classif('3gauss', ns1 + 1)
- transp_Xs_new = otda.transform(Xs_new)
-
- # check that the oos method is working
- assert_equal(transp_Xs_new.shape, Xs_new.shape)
-
-
def test_emd_laplace_class():
"""test_emd_laplace_transport
"""
@@ -654,4 +600,4 @@ def test_emd_laplace_class():
# test fit_transform
transp_Xs = otda.fit_transform(Xs=Xs, Xt=Xt)
- assert_equal(transp_Xs.shape, Xs.shape)
+ assert_equal(transp_Xs.shape, Xs.shape) \ No newline at end of file