├── README.md ├── pic └── stat.png ├── requirements.txt ├── script ├── train_cross_encoder.sh └── train_dual_encoder.sh └── src ├── convert2trec.py ├── dataset_factory.py ├── modeling.py ├── msmarco_eval.py ├── train_cross_encoder.py ├── train_dual_encoder.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUIR/T2Ranking/HEAD/README.md -------------------------------------------------------------------------------- /pic/stat.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUIR/T2Ranking/HEAD/pic/stat.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUIR/T2Ranking/HEAD/requirements.txt -------------------------------------------------------------------------------- /script/train_cross_encoder.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUIR/T2Ranking/HEAD/script/train_cross_encoder.sh -------------------------------------------------------------------------------- /script/train_dual_encoder.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUIR/T2Ranking/HEAD/script/train_dual_encoder.sh -------------------------------------------------------------------------------- /src/convert2trec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUIR/T2Ranking/HEAD/src/convert2trec.py -------------------------------------------------------------------------------- /src/dataset_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUIR/T2Ranking/HEAD/src/dataset_factory.py -------------------------------------------------------------------------------- /src/modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUIR/T2Ranking/HEAD/src/modeling.py -------------------------------------------------------------------------------- /src/msmarco_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUIR/T2Ranking/HEAD/src/msmarco_eval.py -------------------------------------------------------------------------------- /src/train_cross_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUIR/T2Ranking/HEAD/src/train_cross_encoder.py -------------------------------------------------------------------------------- /src/train_dual_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUIR/T2Ranking/HEAD/src/train_dual_encoder.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/THUIR/T2Ranking/HEAD/src/utils.py --------------------------------------------------------------------------------