示例#1
0
 def load_RBM(self, file_path, layer_sizes):
     '''
     load the autoencoder via the RBMs
     '''
     RBM = Autoencoder(layer_sizes)
     RBM = RBM.pretrained_from_file(file_path)
     self.model, self.encoder, self.decoder = RBM.unroll()
     self.set_compiler()
     self.encoder.compile(loss={'encoded': self.kl_loss},
                          optimizer=Adam(self.lr))