Skip to content

Implementation of conversational QA model: FlowQA (with slight improvement)

Notifications You must be signed in to change notification settings

ZhangWen0629/FlowQA

 
 

Repository files navigation

FlowQA

This is our first attempt to make state-of-the-art single-turn QA models conversational. Feel free to build on top of our code to build an even stronger conversational QA model.

For more details, please see: FlowQA: Grasping Flow in History for Conversational Machine Comprehension

Step 1:

perform the following:

pip install -r requirements.txt

to install all dependent python packages.

Step 2:

download necessary files using:

./download.sh

Step 3:

preprocess the data files using:

python preprocess_QuAC.py
python preprocess_CoQA.py

Step 4:

run the training code using:

python train_QuAC.py
python train_CoQA.py

For naming the output model, you can do

python train_OOOO.py --name XXX

Remove any answer marking by:

python train_OOOO.py --explicit_dialog_ctx 0

OOOO is the name of the dataset (QuAC or CoQA).

Step 5:

Do prediction with answer thresholding using

python predict_OOOO.py -m models_XXX/best_model.pt --show SS

XXX is the name you used during train.py.
SS is the number of dialog examples to be shown.
OOOO is the name of the dataset (QuAC or CoQA).

About

Implementation of conversational QA model: FlowQA (with slight improvement)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.5%
  • Shell 0.5%