def build_model(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_v_type=self.rbm_v_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_model()
self.rbm_list.append(rbm) # 加入list
