Skip to content

techit-limtiyakul/bidirectional-attention-flow

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PyTorch implementation for Bi-directional Attention Flow model

Question answering system based on the paper https://arxiv.org/abs/1611.01603. This is a slightly modified version where its answer selection module use bilinear function, giving slight improvement in accuracy over the original model. Part of the code are from https://github.com/jojonki/BiDAF.

In Order to run,

1. Download SQuAD dataset and GloVe embeddings

(~850 MB, this will download files to $HOME/data):

chmod +x download.sh; ./download.sh

2. Preprocess SQuAD data

python -m squad.prepro

Then place the processed data and unzipped GloVe embeddings into the data directory (by default it is ./data/squad)

3. Training

To train, run the following command.

python main.py

To test,

python main.py --test 1 --resume <PATH_TO_SAVED_PARAMS>

About

Implementation of Bi-directional Attention Flow (BiDAF) network https://arxiv.org/abs/1611.01603

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published