model.py 文件源码

def __init__(self,sae,discriminator):
        _data        = Input(shape=(sae.parameters['N'],))
        _data2       = Reshape((sae.parameters['N'],1))(_data)
        _categorical = wrap(_data,K.concatenate([_data2, 1-_data2],-1),name="categorical")
        _images      = sae.decoder(_categorical)
        _features    = sae.features(_images)
        _results     = discriminator.net(_features)
        m            = Model(_data, _results)
        self.model = m

# action autoencoder ################################################################