dmlc--dgl
Gan Quan
ac660f45ef
* [Doc] updated readme * contribution guide change * small edits * add perf # * Update README.md * Update README.md * add others... * rewrite scalability * model & code links * some note for treelstm * rewrite scalability
864 B
864 B
Tree-LSTM
This is a re-implementation of the following paper:
Improved Semantic Representations From Tree-Structured Long Short-Term Memory Networks Kai Sheng Tai, Richard Socher, and Christopher Manning.
The provided implementation can achieve a test accuracy of 51.72 which is comparable with the result reported in the original paper: 51.0(±0.5).
Data
The script will download the [SST dataset] (http://nlp.stanford.edu/sentiment/index.html) automatically, and you need to download the GloVe word vectors yourself. For the command line, you can use this.
wget http://nlp.stanford.edu/data/glove.840B.300d.zip
unzip glove.840B.300d.zip
Usage
python train.py --gpu 0
Speed
On AWS p3.2x instance, it can achieve 3.18s per epoch when setting batch size to 256.