UniRank
UniRank: Benchmarking Ranking Models for Unified Sequential Modeling and Feature Interaction
File Explorer
Download Latest Version (.zip)- test_pipeline.png
- training_pipeline.png
- unirank_logo.png
- EST_KuaiRand_Video_Action.log
- HeMix_KuaiRand_Video_Action.log
- HiFormer_KuaiRand_Video_Action.log
- HyFormer_KuaiRand_Video_Action.log
- INFNet_KuaiRand_Video_Action.log
- LONGER_KuaiRand_Video_Action.log
- MixFormer_KuaiRand_Video_Action.log
- OneTrans_KuaiRand_Video_Action.log
- OneTrans_KuaiRand_Video_Action_AttentionActivation_GeLU.log
- OneTrans_KuaiRand_Video_Action_AttentionActivation_Mish.log
- OneTrans_KuaiRand_Video_Action_AttentionActivation_None.log
- OneTrans_KuaiRand_Video_Action_AttentionActivation_ReLU.log
- OneTrans_KuaiRand_Video_Action_AttentionActivation_Sigmoid.log
- OneTrans_KuaiRand_Video_Action_AttentionActivation_SiLU.log
- OneTrans_KuaiRand_Video_Action_AttentionActivation_SoftPlus.log
- OneTrans_KuaiRand_Video_Action_Tokenizer_Chunk.log
- OneTrans_KuaiRand_Video_Action_Tokenizer_Field.log
- OneTrans_KuaiRand_Video_Action_Tokenizer_Random.log
- OneTrans_var_KuaiRand_Video_Action_AttGate_True.log
- OneTrans_var_KuaiRand_Video_Action_AttRes_True.log
- OneTrans_var_KuaiRand_Video_Action_GNorm_True.log
- OneTrans_var_KuaiRand_Video_Action_QKNorm_True.log
- OneTrans_var_KuaiRand_Video_Action_RoPE_True.log
- OneTrans_var_KuaiRand_Video_Action_SpecT_True.log
- OneTrans_var_KuaiRand_Video_Action_Tau_True.log
- RankMixer_KuaiRand_Video_Action_Optimizer_LaProp.log
- RankMixer_KuaiRand_Video_Action_Optimizer_Muon.log
- RankMixer_KuaiRand_Video_Action_Optimizer_RMSProp.log
- RankMixer_KuaiRand_Video_Action_Optimizer_Scion.log
- RankMixer_KuaiRand_Video_Action_Optimizer_SOAP.log
- RankMixer_KuaiRand_Video_Action_Small.log
- RankMixer_KuaiRand_Video_Action_Tokenizer_Auto.log
- RankMixer_KuaiRand_Video_Action_Tokenizer_Field.log
- RankMixer_KuaiRand_Video_Action_Tokenizer_Random.log
- SSR_KuaiRand_Video_Action.log
- TokenFormer_KuaiRand_Video_Action.log
- TokenMixer_KuaiRand_Video_Action.log
- UltraHSTU_KuaiRand_Video_Action.log
- UniMixer_KuaiRand_Video_Action.log
- Zenith_KuaiRand_Video_Action.log
- EST_MerRec_Action.log
- HeMix_MerRec_Action.log
- HiFormer_MerRec_Action.log
- HyFormer_MerRec_Action.log
- INFNet_MerRec_Action.log
- LONGER_MerRec_Action.log
- MixFormer_MerRec_Action.log
- OneTrans_MerRec_Action.log
- OneTrans_MerRec_Action_AttentionActivation_GeLU.log
- OneTrans_MerRec_Action_AttentionActivation_Mish.log
- OneTrans_MerRec_Action_AttentionActivation_None.log
- OneTrans_MerRec_Action_AttentionActivation_ReLU.log
- OneTrans_MerRec_Action_AttentionActivation_Sigmoid.log
- OneTrans_MerRec_Action_AttentionActivation_SiLU.log
- OneTrans_MerRec_Action_AttentionActivation_SoftPlus.log
- OneTrans_MerRec_Action_Tokenizer_Chunk.log
- OneTrans_MerRec_Action_Tokenizer_Field.log
- OneTrans_MerRec_Action_Tokenizer_Random.log
- OneTrans_var_MerRec_Action_AttGate_True.log
- OneTrans_var_MerRec_Action_AttRes_True.log
- OneTrans_var_MerRec_Action_GNorm_True.log
- OneTrans_var_MerRec_Action_QKNorm_True.log
- OneTrans_var_MerRec_Action_RoPE_True.log
- OneTrans_var_MerRec_Action_SpecT_True.log
- OneTrans_var_MerRec_Action_Tau_True.log
- RankMixer_MerRec_Action_Optimizer_LaProp.log
- RankMixer_MerRec_Action_Optimizer_Muon.log
- RankMixer_MerRec_Action_Optimizer_RMSProp.log
- RankMixer_MerRec_Action_Optimizer_Scion.log
- RankMixer_MerRec_Action_Optimizer_SOAP.log
- RankMixer_MerRec_Action_Small.log
- RankMixer_MerRec_Action_Tokenizer_Auto.log
- RankMixer_MerRec_Action_Tokenizer_Field.log
- RankMixer_MerRec_Action_Tokenizer_Random.log
- SSR_MerRec_Action.log
- TokenFormer_MerRec_Action.log
- TokenMixer_MerRec_Action.log
- UltraHSTU_MerRec_Action.log
- UniMixer_MerRec_Action.log
- Zenith_MerRec_Action.log
- EST_QK_Video_Action.log
- HeMix_QK_Video_Action.log
- HiFormer_QK_Video_Action.log
- HyFormer_QK_Video_Action.log
- INFNet_QK_Video_Action.log
- LONGER_QK_Video_Action.log
- MixFormer_QK_Video_Action.log
- OneTrans_QK_Video_Action.log
- RankMixer_QK_Video_Action.log
- SSR_QK_Video_Action.log
- TokenFormer_QK_Video_Action.log
- TokenMixer_QK_Video_Action.log
- UltraHSTU_QK_Video_Action.log
- UniMixer_QK_Video_Action.log
- Zenith_QK_Video_Action.log
- OneTrans_KuaiRand_Video_Action_Ablation_MaxLen_Large.log
- OneTrans_KuaiRand_Video_Action_Ablation_MaxLen_Mid.log
- OneTrans_KuaiRand_Video_Action_Ablation_MaxLen_Small.log
- OneTrans_KuaiRand_Video_Action_Ablation_MaxLen_Tiny.log
- OneTrans_KuaiRand_Video_Action_Ablation_NumLayers_Large.log
- OneTrans_KuaiRand_Video_Action_Ablation_NumLayers_Mid.log
- OneTrans_KuaiRand_Video_Action_Ablation_NumLayers_Small.log
- OneTrans_KuaiRand_Video_Action_Ablation_TokenDim_Mid.log
- OneTrans_KuaiRand_Video_Action_Ablation_TokenDim_Small.log
- OneTrans_KuaiRand_Video_Action_Ablation_TokenDim_Tiny.log
- OneTrans_KuaiRand_Video_Action_Ablation_TokenDim_Ultra.log
- RankMixer_KuaiRand_Video_Action_Ablation_MaxLen_Large.log
- RankMixer_KuaiRand_Video_Action_Ablation_MaxLen_Mid.log
- RankMixer_KuaiRand_Video_Action_Ablation_MaxLen_Small.log
- RankMixer_KuaiRand_Video_Action_Ablation_MaxLen_Tiny.log
- RankMixer_KuaiRand_Video_Action_Ablation_NumLayers_Large.log
- RankMixer_KuaiRand_Video_Action_Ablation_NumLayers_Mid.log
- RankMixer_KuaiRand_Video_Action_Ablation_TokenDim_Large.log
- RankMixer_KuaiRand_Video_Action_Ablation_TokenDim_Mid.log
- RankMixer_KuaiRand_Video_Action_Ablation_TokenDim_Small.log
- RankMixer_KuaiRand_Video_Action_Ablation_TokenDim_Tiny.log
- RankMixer_KuaiRand_Video_Action_Ablation_TokenDim_Ultra.log
- EST_Taobao_Action.log
- HeMix_Taobao_Action.log
- HiFormer_Taobao_Action.log
- HyFormer_Taobao_Action.log
- INFNet_Taobao_Action.log
- LONGER_Taobao_Action.log
- MixFormer_Taobao_Action.log
- OneTrans_Taobao_Action.log
- RankMixer_Taobao_Action.log
- SSR_Taobao_Action.log
- TokenFormer_Taobao_Action.log
- TokenMixer_Taobao_Action.log
- UltraHSTU_Taobao_Action.log
- UniMixer_Taobao_Action.log
- Zenith_Taobao_Action.log
- EST_TencentGR_10M_Action.log
- HeMix_TencentGR_10M_Action.log
- HiFormer_TencentGR_10M_Action.log
- HyFormer_TencentGR_10M_Action.log
- INFNet_TencentGR_10M_Action.log
- LONGER_TencentGR_10M_Action.log
- MixFormer_TencentGR_10M_Action.log
- OneTrans_TencentGR_10M_Action.log
- RankMixer_TencentGR_10M_Action.log
- SSR_TencentGR_10M_Action.log
- TokenFormer_TencentGR_10M_Action.log
- TokenMixer_TencentGR_10M_Action.log
- UltraHSTU_TencentGR_10M_Action.log
- UniMixer_TencentGR_10M_Action.log
- Zenith_TencentGR_10M_Action.log
- plot_scaling_auc_vs_gflops.py
- dataset_config.yaml
- model_config.yaml
- download.sh
- preprocess_Kuairand_seq_action.py
- stat_KuaiRand_dataset.py
- download_merrec_20231001.py
- preprocess_MerRec_seq_action.py
- stat_MerRec_dataset.py
- preprocess_QK_seq_action.py
- stat_QK_dataset.py
- export_parquet.py
- preprocess_TAAC2025_seq_action.py
- stat_TAAC2025_dataset.py
- preprocess_Taobao_seq_action.py
- stat_Taobao_dataset.py
- dataset_stats_utils.py
- __init__.py
- EST.py
- HeMix.py
- HiFormer.py
- HyFormer.py
- INFNet.py
- LONGER.py
- MixFormer.py
- OneTrans.py
- RankMixer.py
- SSR.py
- TokenFormer.py
- TokenMixer.py
- UltraHSTU.py
- UniMixer.py
- Zenith.py
- __init__.py
- build_dataset.py
- feature_processor.py
- normalizer.py
- tokenizer.py
- __init__.py
- rank_dataloader.py
- unirank_dataloader.py
- __init__.py
- dot_product_attention.py
- target_attention.py
- __init__.py
- mlp_block.py
- __init__.py
- feature_embedding.py
- pretrained_embedding.py
- __init__.py
- per_token_ffn.py
- token_mix.py
- __init__.py
- auto_split_tokenizer.py
- chunk_tokenizer.py
- field_wise_tokenizer.py
- random_split_tokenizer.py
- unified_tokenizer.py
- __init__.py
- activations.py
- pooling.py
- __init__.py
- multitask_model.py
- rank_model.py
- __init__.py
- torch_utils.py
- __init__.py
- features.py
- metrics.py
- utils.py
- .gitignore
- autotuner.py
- LICENSE
- README.md
- requirements.txt
- run_all.sh
- run_expid.py
- run_param_tuner.py
# Installation Guide
1. Get the code
git clone https://github.com/salmon1802/UniRank
Downloads the entire project code from GitHub to your computer.
cd UniRank
Moves into the project folder you just downloaded.
2. Official Install Script
Easy RecommendedPrerequisites
- Python 3 Python is required to use pip.
pip install torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 \
Installs the package published on PyPI directly β no need to clone the source.
After installing, open a new terminal and run the program's version command (e.g. --version) to confirm it worked.
Pulled directly from this repo's README.
3. Python
EasyPrerequisites
pip install torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 \
Installs the package published on PyPI directly β no need to clone the source.
pip install -r requirements.txt
Installs the Python libraries listed in requirements.txt (or similar).
If it runs without errors and prints output in the terminal, it worked.
Pulled directly from this repo's README.
// repository documentation
Was this content helpful?
(0 ratings)