Skip to content

Commit 6330256

Browse files
author
ysmiao
committed
neural variational document model
0 parents  commit 6330256

File tree

6 files changed

+21103
-0
lines changed

6 files changed

+21103
-0
lines changed

README.md

+25
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,25 @@
1+
# Neural Variational Document Model (NVDM) tensorflow implementation
2+
3+
------
4+
5+
This is the tensorflow implementation of NVDM for the paper:
6+
[Neural Variational Inference for Text Processing][1]. Yishu Miao, Lei Yu, Phil Blunsom. ICML 2016.
7+
8+
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.
9+
10+
### RCV1-v2 dataset
11+
Please download and uncompress the [dataset][2] to:
12+
```
13+
data/rcv1-v2
14+
```
15+
### Train the Model
16+
17+
```
18+
python nvdm.py --data_dir data/20news/
19+
```
20+
21+
22+
23+
[1]: https://arxiv.org/abs/1511.06038
24+
[2]: https://drive.google.com/open?id=0ByuyRPR04lmnVzhmT1RoVTZOb1E
25+
[3]: https://arxiv.org/abs/1511.06038

data/20news/test.feat

+7,505
Large diffs are not rendered by default.

data/20news/train.feat

+11,268
Large diffs are not rendered by default.

0 commit comments

Comments
 (0)