├── .gitignore ├── README.md ├── images └── tensorboard_ndcg.png ├── ranking ├── LambdaRank.py ├── ListNet.py ├── RankNet.py ├── __init__.py ├── data_loaders │ ├── __init__.py │ └── load_expedia.py ├── download_data.sh ├── load_mslr.py ├── metrics.py ├── positional_bias.py └── utils.py ├── transformer ├── dataloader_fineweb.py ├── dataset_utils.py ├── decoder_mha_gqa.py ├── rotary_embedding.py ├── train_decoder_simple.py └── transformer_encoder_decoder.py └── uplift_model ├── __init__.py ├── eval_uplift_effect.py └── rank_uplift.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/README.md -------------------------------------------------------------------------------- /images/tensorboard_ndcg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/images/tensorboard_ndcg.png -------------------------------------------------------------------------------- /ranking/LambdaRank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/ranking/LambdaRank.py -------------------------------------------------------------------------------- /ranking/ListNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/ranking/ListNet.py -------------------------------------------------------------------------------- /ranking/RankNet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/ranking/RankNet.py -------------------------------------------------------------------------------- /ranking/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ranking/data_loaders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /ranking/data_loaders/load_expedia.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/ranking/data_loaders/load_expedia.py -------------------------------------------------------------------------------- /ranking/download_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/ranking/download_data.sh -------------------------------------------------------------------------------- /ranking/load_mslr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/ranking/load_mslr.py -------------------------------------------------------------------------------- /ranking/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/ranking/metrics.py -------------------------------------------------------------------------------- /ranking/positional_bias.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/ranking/positional_bias.py -------------------------------------------------------------------------------- /ranking/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/ranking/utils.py -------------------------------------------------------------------------------- /transformer/dataloader_fineweb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/transformer/dataloader_fineweb.py -------------------------------------------------------------------------------- /transformer/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/transformer/dataset_utils.py -------------------------------------------------------------------------------- /transformer/decoder_mha_gqa.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/transformer/decoder_mha_gqa.py -------------------------------------------------------------------------------- /transformer/rotary_embedding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/transformer/rotary_embedding.py -------------------------------------------------------------------------------- /transformer/train_decoder_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/transformer/train_decoder_simple.py -------------------------------------------------------------------------------- /transformer/transformer_encoder_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/transformer/transformer_encoder_decoder.py -------------------------------------------------------------------------------- /uplift_model/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /uplift_model/eval_uplift_effect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/uplift_model/eval_uplift_effect.py -------------------------------------------------------------------------------- /uplift_model/rank_uplift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/haowei01/pytorch-examples/HEAD/uplift_model/rank_uplift.py --------------------------------------------------------------------------------