https://github.com/pl8787/setrank
Implementation of SetRank in SIGIR 2020
https://github.com/pl8787/setrank
information-retrieval learning-to-rank
Last synced: 6 months ago
JSON representation
Implementation of SetRank in SIGIR 2020
- Host: GitHub
- URL: https://github.com/pl8787/setrank
- Owner: pl8787
- Created: 2020-04-24T08:00:04.000Z (over 5 years ago)
- Default Branch: master
- Last Pushed: 2020-04-25T05:24:04.000Z (over 5 years ago)
- Last Synced: 2025-03-26T11:21:46.686Z (6 months ago)
- Topics: information-retrieval, learning-to-rank
- Language: Python
- Homepage:
- Size: 188 KB
- Stars: 50
- Watchers: 1
- Forks: 7
- Open Issues: 1
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# SetRank
> Tenforflow implementation of [SIGIR 2020] SetRank: Learning a Permutation-Invariant Ranking Model for Information Retrieval.
[](https://www.python.org/downloads/release/python-360/)
[](https://opensource.org/licenses/Apache-2.0)
## Usage
### Quick Start
> If you have downloaded **Istella Dataset** please comment `sh download_data.sh`.
```bash
sh run.sh
```
### Data Preparation
```bash
cd data
sh download_data.sh
python norm_split_dataset.py
```
### Model Training/Testing
```bash
sh ./scripts/train_lambdamart_istella.sh
sh ./scripts/prepare_data_lambda_istella.sh
sh ./scripts/train_transformer_istella.sh
```
## Citation
If you use SetRank in your research, please use the following BibTex entry.
```
@inproceedings{pang2019setrank,
title={SetRank: Learning a Permutation-Invariant Ranking Model for Information Retrieval},
author={Liang Pang and Jun Xu and Qingyao Ai and Yanyan Lan and Xueqi Cheng and Jirong Wen},
booktitle = {Proceedings of the 43rd International ACM SIGIR Conference on Research and Development in Information Retrieval},
series = {SIGIR'20},
year = {2020},
}
```
## License
[Apache-2.0](https://opensource.org/licenses/Apache-2.0)
Copyright (c) 2019-present, Liang Pang (pl8787)