Source code of SIGIR2020 Long Paper "Encoding History with Context-aware Representation Learning for Personalized Search". Due to two sub-models QDM and PLM have similar structures, we release the code of QDM model here which has a more classic hierarchical transformer structure.
I test the code with the following packages. Other versions may also work, but I'm not sure.
- Python 3.6
- Pytorch 1.3.1 (with GPU support)
Data processing and train the model.
python dataset_new.py
If you use the code, please cite the following paper:
@inproceedings{ZhouDW20,
author = {Yujia Zhou and
Zhicheng Dou and
Ji{-}Rong Wen},
title = {Encoding History with Context-aware Representation Learning for Personalized
Search},
booktitle = {{SIGIR}},
pages = {1111--1120},
publisher = {{ACM}},
year = {2020}
}