コード例 #1
0
def test_sk_ae():
    sk = AutoEncoder(layers_dim=[
        20,
    ], encoding_dim=2)
    sk3 = AutoEncoder(layers_dim=[
        20,
    ], encoding_dim=10)

    X2 = sk.fit_transform(X)
    X3 = sk3.fit_transform(X)

    si = silhouette_score(X2, y)
    assert si > 0.8, 'Score {} < 0.8'.format(si)
    print('X2 Silhouette score: {}'.format(si))

    Xr = sk.inverse_transform(X2)
    error = mse(X, Xr)
    print('Error : {}'.format(error))
    Xr3 = sk3.inverse_transform(X3)
    error3 = mse(X, Xr3)
    print('Error3 : {}'.format(error3))
    assert error3 < error, 'Bigger network has more error'