├── .gitignore ├── README.md ├── data ├── download_data.sh └── norm_split_dataset.py ├── ranklib └── RankLib-2.12.jar ├── requirements.txt ├── run.sh ├── scripts ├── evaluate_istella.sh ├── prepare_data_lambda_istella.sh ├── prepare_istella_data.py ├── train_lambdamart_istella.sh └── train_transformer_istella.sh └── src ├── RankTransformer_model.py ├── data_utils.py ├── main_transformer.py └── modules.py /.gitignore: -------------------------------------------------------------------------------- 1 | processed_data 2 | *.log 3 | *.pyc 4 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pl8787/SetRank/HEAD/README.md -------------------------------------------------------------------------------- /data/download_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pl8787/SetRank/HEAD/data/download_data.sh -------------------------------------------------------------------------------- /data/norm_split_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pl8787/SetRank/HEAD/data/norm_split_dataset.py -------------------------------------------------------------------------------- /ranklib/RankLib-2.12.jar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pl8787/SetRank/HEAD/ranklib/RankLib-2.12.jar -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pl8787/SetRank/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pl8787/SetRank/HEAD/run.sh -------------------------------------------------------------------------------- /scripts/evaluate_istella.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pl8787/SetRank/HEAD/scripts/evaluate_istella.sh -------------------------------------------------------------------------------- /scripts/prepare_data_lambda_istella.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pl8787/SetRank/HEAD/scripts/prepare_data_lambda_istella.sh -------------------------------------------------------------------------------- /scripts/prepare_istella_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pl8787/SetRank/HEAD/scripts/prepare_istella_data.py -------------------------------------------------------------------------------- /scripts/train_lambdamart_istella.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pl8787/SetRank/HEAD/scripts/train_lambdamart_istella.sh -------------------------------------------------------------------------------- /scripts/train_transformer_istella.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pl8787/SetRank/HEAD/scripts/train_transformer_istella.sh -------------------------------------------------------------------------------- /src/RankTransformer_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pl8787/SetRank/HEAD/src/RankTransformer_model.py -------------------------------------------------------------------------------- /src/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pl8787/SetRank/HEAD/src/data_utils.py -------------------------------------------------------------------------------- /src/main_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pl8787/SetRank/HEAD/src/main_transformer.py -------------------------------------------------------------------------------- /src/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pl8787/SetRank/HEAD/src/modules.py --------------------------------------------------------------------------------