├── .gitignore ├── CE ├── README.md ├── models │ └── ranker.py ├── re-rank.py ├── train_ranker.py └── utils │ ├── data_utils.py │ ├── dist_utils.py │ ├── model_utils.py │ └── options.py ├── LICENSE ├── README.md ├── data.py ├── data ├── msmarco_data │ ├── dev.query.tsv │ ├── qrels.dev.small.tsv │ ├── qrels.train.tsv │ └── train.query.tsv ├── process_marco.py └── process_xorqa.py ├── get_data.sh ├── requirements.txt ├── run.py └── trainer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/.gitignore -------------------------------------------------------------------------------- /CE/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/CE/README.md -------------------------------------------------------------------------------- /CE/models/ranker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/CE/models/ranker.py -------------------------------------------------------------------------------- /CE/re-rank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/CE/re-rank.py -------------------------------------------------------------------------------- /CE/train_ranker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/CE/train_ranker.py -------------------------------------------------------------------------------- /CE/utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/CE/utils/data_utils.py -------------------------------------------------------------------------------- /CE/utils/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/CE/utils/dist_utils.py -------------------------------------------------------------------------------- /CE/utils/model_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/CE/utils/model_utils.py -------------------------------------------------------------------------------- /CE/utils/options.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/CE/utils/options.py -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/README.md -------------------------------------------------------------------------------- /data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/data.py -------------------------------------------------------------------------------- /data/msmarco_data/dev.query.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/data/msmarco_data/dev.query.tsv -------------------------------------------------------------------------------- /data/msmarco_data/qrels.dev.small.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/data/msmarco_data/qrels.dev.small.tsv -------------------------------------------------------------------------------- /data/msmarco_data/qrels.train.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/data/msmarco_data/qrels.train.tsv -------------------------------------------------------------------------------- /data/msmarco_data/train.query.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/data/msmarco_data/train.query.tsv -------------------------------------------------------------------------------- /data/process_marco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/data/process_marco.py -------------------------------------------------------------------------------- /data/process_xorqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/data/process_xorqa.py -------------------------------------------------------------------------------- /get_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/get_data.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/run.py -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ArvinZhuang/DSI-QG/HEAD/trainer.py --------------------------------------------------------------------------------