This is an implementation of this paper(CORE: Simple and Effective Session-based Recommendation within Consistent Representation Space) based on Pytorch 2.X, which contains the extra functions as below:
- Log(output for loss and MRR@20, P@20)
- train & test progress bar display
- preformance enhanced evaluation for metrics
Pytorch 2.X (version>=1.8 is compatible)
Python 3.10 (version>=3.8 is compatible)
CUDA 11.8 and above is prefer
cudnn 8.7.0 and above is prefer
Citation
@inproceedings{wang2020global,
title={Global Context Enhanced Graph Neural Networks for Session-based Recommendation},
author={Wang, Ziyang and Wei, Wei and Cong, Gao and Li, Xiao-Li and Mao, Xian-Ling and Qiu, Minghui},
booktitle={Proceedings of the 43rd International ACM SIGIR Conference on Research and Development in Information Retrieval},
pages={169--178},
year={2020}
}