Skip to content
/ nvdm Public
forked from ysmiao/nvdm

neural variational document model

Notifications You must be signed in to change notification settings

skesiraju/nvdm

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 

Repository files navigation

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/

About

neural variational document model

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%