gated-attention-reader
gated-attention-reader copied to clipboard
Tensorflow/Pytorch implementation of Gated Attention Reader
GA-Reader
Tensorflow implementation of Gated Attention Reader for Text Comprehension. The original code can be found from here. For pytorch implementation, please check pytorch branch.
Prerequisites
- Python 3.5
- TensorFlow 1.0+
- tqdm
Preprocessed Data
You can get the preprocessed data files from here.
You can also get the pretrained Glove vectors from the above link.
Training
python main.py --data_dir ~/data/dailymail --embed_file ~/data/word2vec_glove.txt
You should see around 0.7 accuracy on training set after running 1000 iteration (total 27486 iterations) on dailymail with default setting.