autoencoders.py 文件源码

def multilayer_autoencoder(X, lam=1e-5):
    X = X.reshape(X.shape[0], -1)
    M, N = X.shape

    inputs = Input(shape=(N,))
    h = Dense(128, activation='relu')(inputs)
    encoded = Dense(64, activation='relu', activity_regularizer=activity_l1(lam))(h)
    h = Dense(128, activation='relu')(encoded)
    outputs = Dense(N)(h)

    model = Model(input=inputs, output=outputs)
    model.compile(optimizer='adam', loss='mse')
    model.fit(X, X, batch_size=64, nb_epoch=3)

    return model, Model(input=inputs, output=h)