Esempio n. 1
1
def test_tga_check_list():
    # Test that the projection by TGA on list data is correct
    X = [[1.0, 0.0], [0.0, 1.0]]
    X_transformed = TGA(n_components=1,
                        random_state=0).fit(X).transform(X)
    assert_equal(X_transformed.shape, (2, 1))
    assert_almost_equal(X_transformed.mean(), 0.00, 2)
    assert_almost_equal(X_transformed.std(), 0.71, 2)