Implementation of conversational QA model: FlowQA (with slight improvement)
Implementation of conversational QA model: FlowQA (with slight improvement)
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
perform the following:
pip install -r requirements.txt
to install all dependent python packages.
download necessary files using:
./download.sh
preprocess the data files using:
python preprocess_QuAC.py
python preprocess_CoQA.py
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).
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).
No open issues yet, or sync has not completed.