Awesome

Neural Variational Document Model (NVDM) tensorflow implementation

This is the tensorflow implementation of NVDM for the paper: Neural Variational Inference for Text Processing. Yishu Miao, Lei Yu, Phil Blunsom. ICML 2016.

The original code is on torch. Since there are quite a few people asked me questions about the implementation, I have reimplemented the model by tensorflow. Please contact me if you find any problem with this implementation. It is able to achieve better results than the ones reported in the paper.

RCV1-v2 dataset

Please download and uncompress the dataset to:

data/rcv1-v2

Train the Model

python nvdm.py --data_dir data/20news/