def build_rbms(self):
# feed 变量
self.input_data = tf.placeholder(
tf.float32,
[None, self.rbms_struct[0]]) # N等于_num_examples或batch_size
# 构建rmbs
self.rbm_list = list()
for i in range(len(self.rbms_struct) - 1):
n_v = self.rbms_struct[i]
n_h = self.rbms_struct[i + 1]
rbm = RBM(rbm_h_type=self.rbm_h_type,
rbm_struct=[n_v, n_h],
rbm_epochs=self.rbm_epochs,
batch_size=self.batch_size,
cd_k=self.cd_k,
rbm_lr=self.rbm_lr)
rbm.build_rbm()
self.rbm_list.append(rbm) # 加入list
