def __init__(self, sess, reader, dataset="ptb",
batch_size=20, num_steps=3, embed_dim=500,
h_dim=50, learning_rate=0.01, epoch=50,
checkpoint_dir="checkpoint"):
"""Initialize Neural Varational Document Model.
params:
sess: TensorFlow Session object.
reader: TextReader object for training and test.
dataset: The name of dataset to use.
h_dim: The dimension of document representations (h). [50, 200]
"""
self.sess = sess
self.reader = reader
self.h_dim = h_dim
self.embed_dim = embed_dim
self.epoch = epoch
self.batch_size = batch_size
self.learning_rate = learning_rate
self.checkpoint_dir = checkpoint_dir
self.dataset="ptb"
self._attrs=["batch_size", "num_steps", "embed_dim", "h_dim", "learning_rate"]
raise Exception(" [!] Working in progress")
self.build_model()
评论列表
文章目录