Skip to content

JD-AI-Research-NLP/MuGER2

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

36 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Main Requirements:

pytorch==1.5.0

transformers==2.6.0

Paper result:

You need to download the WIKI-LINKS, and put it in ./processed_data/.

Unzip the WIKI-LINKS and the retrieval_score_large.zip.

Download the trained MRC models to folder ./trained_model/. <提取码:tftr>

Download the pre-trained BERT-large model to folder ./muger2/bertlarge/.

After preparing the trained models, the paper results can be obtained through the following commands.

cd muger2
python e-selector.py --mode dev --mrc_mode large --input_path ../processed_data/retrieval_scores_large.json --output_path predictions_large.json
python evaluate_script.py predictions_large.json dev_reference.json

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages