Need help with BiDAF?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

jojonki
139 Stars 28 Forks Apache License 2.0 121 Commits 4 Opened issues

Description

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

Services available

!
?

Need anything else?

Contributors list

# 297,747
Python
Shell
dialogu...
convers...
103 commits

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

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.