chunlinx / bidaf

Bidirectional Attention Flow for Machine Comprehension, Minjoon Seo, Aniruddha Kembhavi, Ali Farhadi, Hannaneh Hajishirzi https://arxiv.org/abs/1611.01603

Bidirectional Attention Flow for Machine Comprehension

Bidirectional Attention Flow for Machine Comprehension, Minjoon Seo, Aniruddha Kembhavi, Ali Farhadi, Hannaneh Hajishirzi https://arxiv.org/abs/1611.01603

This repo contains the implementation of Bi-Directional Attention Flow (BIDAF) network in PyTorch.

Requirements

• PyTorch: 0.3
• tqdm (progress bar)

Setup

$ ./download.sh  // download squad dataset
$ python -m squad.prepro    // build documents

And you also need to download glove 6B dataset under dataset directory.

Training

$ python main.py --help
$ python main.py

Test

$ python main.py --resume ./checkpoints/Epoch-12.model --test 1

TODO

• confirm this model's performance, currently performance does not increasing. (see #1 )
• Support multi labels
• Test. The answer span (k, l) where k ≤ l with the maximum value of (p1k, p2l) should be chosen.
• Imple EM and F1 score
• clean main.py