Esempio n. 1
1
def test_tga_median():
    # TGA on dense arrays
    tga = TGA(n_components=2, random_state=1, centering='median')
    X = iris.data
    X_r = tga.fit(X).transform(X)
    assert_equal(X_r.shape[1], 2)

    tga = TGA(n_components=2, random_state=1, centering='median')
    X_r2 = tga.fit_transform(X)
    assert_array_almost_equal(X_r, X_r2)

    tga = TGA(random_state=1, centering='median')
    tga.fit(X)
    X_r = tga.transform(X)
    tga = TGA(random_state=1, centering='median')
    X_r2 = tga.fit_transform(X)
    assert_array_almost_equal(X_r, X_r2)