From d6ef8676cc3f94ba5d80acc9fd9745c9ed91819a Mon Sep 17 00:00:00 2001 From: ievred Date: Wed, 8 Apr 2020 10:28:57 +0200 Subject: remove jcpot from laplace --- test/test_da.py | 56 +------------------------------------------------------- 1 file changed, 1 insertion(+), 55 deletions(-) (limited to 'test') 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 -- cgit v1.2.3