def init_train_args(self, mode='recode'):
# To be used for training by tf.Optimizer objects.
self.train_args = [tf.placeholder(self.dtype,
shape=[None, self.n_visible])]
if mode == 'target':
self.train_args.append(tf.placeholder(self.dtype,
shape=[None, self.n_hidden]))
elif mode == 'label':
self.train_args.append(tf.placeholder(tf.int32, shape=[None]))
return self.train_args
评论列表
文章目录