SQuAD-tensorflow
Implementation of several models for SQuAD
- Basic LSTM based Model
- Multi-Perspective Context Matching (IBM)
- BiDAF
Requirements
- Python 3.4.3 or higher
- Tensorflow 1.0.1 (GPU enabled)
- Cuda settings
- Gensim
Downloads
- ./data/ : SQuAD dataset
- ~/embed_data : glove pretrained file
Run the code
# model: b (Basic), m (MPCM)
# debug: shows prediction process
# test: only runs few initial iterations
$ python main --model m --debug True --test False