U-SetRank

Tenforflow implementation of [CCIR 2022] U-SetRank：基于注意力机制修正的 SetRank 纠偏。

Usage

Environmental preparation

you should first download galago and put it in \scripts.

tf 1.0 and tensorboardX are also used

Quick Start

If you have downloaded Istella Dataset please comment sh download_data.sh.

sh test.sh

Data Preparation

cd data
sh download_data.sh
python norm_split_dataset.py

Model Training/Testing

sh ./scripts/train_lambdamart_istella.sh
sh ./scripts/prepare_data_lambda_istella.sh
sh ./scripts/train_transformer_istella.sh

Datasets

Istella Dataset

More information in https://tensorflow.google.cn/datasets/catalog/istella

License

Apache-2.0