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'