Перейти к файлу
Chen Zhao 953d8d4167
Merge pull request #1 from maxbachmann-forks/master
use rapidfuzz instead of fuzzywuzzy
2020-10-27 08:49:42 -04:00
transformer-xh use rapidfuzz instead of fuzzywuzzy 2020-03-28 22:52:27 +01:00
.gitignore upload code for initial release 2020-03-09 21:18:53 -07:00
CODE_OF_CONDUCT.md Initial CODE_OF_CONDUCT.md commit 2020-01-29 11:32:40 -08:00
CONTRIBUTING.md Create CONTRIBUTING.md 2020-02-04 14:28:54 -08:00
Evaluator.py small fix, update readme 2020-04-27 21:12:56 -04:00
LICENSE Initial LICENSE commit 2020-01-29 11:32:42 -08:00
LICENSE.txt upload code for initial release 2020-03-09 21:18:53 -07:00
README.md small fix, update readme 2020-04-27 21:12:56 -04:00
SECURITY.md Initial SECURITY.md commit 2020-01-29 11:32:45 -08:00
Trainer.py small fix, update readme 2020-04-27 21:12:56 -04:00
download.sh upload code for initial release 2020-03-09 21:18:53 -07:00
fever_eval.sh upload code for initial release 2020-03-09 21:18:53 -07:00
fever_train.sh upload code for initial release 2020-03-09 21:18:53 -07:00
hotpot_eval.sh upload code for initial release 2020-03-09 21:18:53 -07:00
hotpot_train.sh upload code for initial release 2020-03-09 21:18:53 -07:00
main.py small fix, update readme 2020-04-27 21:12:56 -04:00
setup.py use rapidfuzz instead of fuzzywuzzy 2020-03-28 22:52:27 +01:00
utils.py use rapidfuzz instead of fuzzywuzzy 2020-03-28 22:52:27 +01:00

README.md

Microsoft Open Source Code of Conduct

This project has adopted the Microsoft Open Source Code of Conduct.

Resources:

Transformer-XH

The source codes of the paper "Transformer-XH: Multi-evidence Reasoning with Extra Hop Attention (ICLR 2020)".

Dependency Installation

First, Run python setup.py develop to install required dependencies for transformer-xh. Also install apex (for distributed training) following official documentation here.

Data and trained model Download

You can run bash script download.sh

For Hotpot QA, we provide processed graph (Transformer-XH) input here, after downloading, unzip it and put into ./data folder We also provide trained model here, unzip the downloaded model and put into ./experiments folder

Similarly, we provide processed graph in fever here, and trained model here.

Run Your Models

Use hotpot_train.sh for training on hotpot QA task, hotpot_eval.sh for evaluation (default fp16 training).

Similarly, fever_train.sh for training on FEVER task, fever_eval.sh for evaluation (default fp16 training).

Contact

If you have questions, suggestions and bug reports, please email chenz@cs.umd.edu and/or Chenyan.Xiong@microsoft.com.