sde_autoencoder.py 文件源码

def create_decoder(input_size, decode_size, weights):
    decoder_layers = [
        (InputLayer, {'shape': (None, input_size)}),
        (DenseLayer, {'name': 'decoder', 'num_units': decode_size, 'nonlinearity': linear, 'W': weights})
    ]

    decoder = NeuralNet(
        layers=decoder_layers,
        max_epochs=50,
        objective_loss_function=squared_error,
        update=adadelta,
        regression=True,
        verbose=1
    )
    return decoder