├── .gitignore ├── Env.py ├── MonteCarlo_test.py ├── MonteCarlo_train.py ├── Readme.md ├── approachs ├── PRMModel.py ├── __init__.py ├── discriminator.py ├── dnn_model.py ├── dqn_model.py ├── groupwise_model.py ├── listwise_model.py ├── lp_model.py ├── model.py ├── pairwise_model.py ├── pointwise_model.py ├── ppo_gae_model.py ├── ppo_model.py ├── rl_model.py ├── rl_sl_model.py └── softrank.py ├── dataset.py ├── dataset ├── test-set.pk ├── train-set.pk └── validation-set.pk ├── dnn_test.py ├── dnn_train.py ├── documents └── documents.npy ├── dqn_test.py ├── dqn_train.py ├── evaluator_train.py ├── model_eval ├── checkpoint ├── ctr_model-291601.data-00000-of-00001 ├── ctr_model-291601.index └── ctr_model-291601.meta ├── ppo_gae_test.py ├── ppo_gae_train.py ├── ppo_plus_test.py ├── ppo_plus_train.py ├── ppo_test.py ├── ppo_train.py ├── prm_test.py ├── prm_train.py ├── seq2slate_test.py ├── seq2slate_train.py ├── test_model.py ├── user ├── checkpoint ├── user-model-0.data-00000-of-00001 ├── user-model-0.index └── user-model-0.meta └── utils ├── __init__.py ├── io_utils.py └── measure.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/.gitignore -------------------------------------------------------------------------------- /Env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/Env.py -------------------------------------------------------------------------------- /MonteCarlo_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/MonteCarlo_test.py -------------------------------------------------------------------------------- /MonteCarlo_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/MonteCarlo_train.py -------------------------------------------------------------------------------- /Readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/Readme.md -------------------------------------------------------------------------------- /approachs/PRMModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/approachs/PRMModel.py -------------------------------------------------------------------------------- /approachs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /approachs/discriminator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/approachs/discriminator.py -------------------------------------------------------------------------------- /approachs/dnn_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/approachs/dnn_model.py -------------------------------------------------------------------------------- /approachs/dqn_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/approachs/dqn_model.py -------------------------------------------------------------------------------- /approachs/groupwise_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/approachs/groupwise_model.py -------------------------------------------------------------------------------- /approachs/listwise_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/approachs/listwise_model.py -------------------------------------------------------------------------------- /approachs/lp_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/approachs/lp_model.py -------------------------------------------------------------------------------- /approachs/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/approachs/model.py -------------------------------------------------------------------------------- /approachs/pairwise_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/approachs/pairwise_model.py -------------------------------------------------------------------------------- /approachs/pointwise_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/approachs/pointwise_model.py -------------------------------------------------------------------------------- /approachs/ppo_gae_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/approachs/ppo_gae_model.py -------------------------------------------------------------------------------- /approachs/ppo_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/approachs/ppo_model.py -------------------------------------------------------------------------------- /approachs/rl_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/approachs/rl_model.py -------------------------------------------------------------------------------- /approachs/rl_sl_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/approachs/rl_sl_model.py -------------------------------------------------------------------------------- /approachs/softrank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/approachs/softrank.py -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/dataset.py -------------------------------------------------------------------------------- /dataset/test-set.pk: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/dataset/test-set.pk -------------------------------------------------------------------------------- /dataset/train-set.pk: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/dataset/train-set.pk -------------------------------------------------------------------------------- /dataset/validation-set.pk: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/dataset/validation-set.pk -------------------------------------------------------------------------------- /dnn_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/dnn_test.py -------------------------------------------------------------------------------- /dnn_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/dnn_train.py -------------------------------------------------------------------------------- /documents/documents.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/documents/documents.npy -------------------------------------------------------------------------------- /dqn_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/dqn_test.py -------------------------------------------------------------------------------- /dqn_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/dqn_train.py -------------------------------------------------------------------------------- /evaluator_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/evaluator_train.py -------------------------------------------------------------------------------- /model_eval/checkpoint: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/model_eval/checkpoint -------------------------------------------------------------------------------- /model_eval/ctr_model-291601.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/model_eval/ctr_model-291601.data-00000-of-00001 -------------------------------------------------------------------------------- /model_eval/ctr_model-291601.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/model_eval/ctr_model-291601.index -------------------------------------------------------------------------------- /model_eval/ctr_model-291601.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/model_eval/ctr_model-291601.meta -------------------------------------------------------------------------------- /ppo_gae_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/ppo_gae_test.py -------------------------------------------------------------------------------- /ppo_gae_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/ppo_gae_train.py -------------------------------------------------------------------------------- /ppo_plus_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/ppo_plus_test.py -------------------------------------------------------------------------------- /ppo_plus_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/ppo_plus_train.py -------------------------------------------------------------------------------- /ppo_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/ppo_test.py -------------------------------------------------------------------------------- /ppo_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/ppo_train.py -------------------------------------------------------------------------------- /prm_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/prm_test.py -------------------------------------------------------------------------------- /prm_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/prm_train.py -------------------------------------------------------------------------------- /seq2slate_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/seq2slate_test.py -------------------------------------------------------------------------------- /seq2slate_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/seq2slate_train.py -------------------------------------------------------------------------------- /test_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/test_model.py -------------------------------------------------------------------------------- /user/checkpoint: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/user/checkpoint -------------------------------------------------------------------------------- /user/user-model-0.data-00000-of-00001: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/user/user-model-0.data-00000-of-00001 -------------------------------------------------------------------------------- /user/user-model-0.index: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/user/user-model-0.index -------------------------------------------------------------------------------- /user/user-model-0.meta: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/user/user-model-0.meta -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/io_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/utils/io_utils.py -------------------------------------------------------------------------------- /utils/measure.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaoyq95/RerankSim/HEAD/utils/measure.py --------------------------------------------------------------------------------