GraphDR
GraphDR is an effective, efficient, and easy-to-deploy matching (i.e., candidate generation in two-stage recommendation) model based on heterogeneous interactions among user, item, category, tag, word, and media. The details of GraphDR are in Improving Accuracy and Diversity in Matching of Recommendation with Diversified Preference Network(TBD-2021) Ruobing Xie*, Qi Liu*, Shukai Liu, Ziwei Zhang, Peng Cui, Bo Zhang and Leyu Lin.
Operating environment
python 2.7.15 tensorflow 1.14.0
Train GraphDR model
sh -x train.sh
CITE
If the codes help you, please cite the following paper:
Ruobing Xie*, Qi Liu*, Shukai Liu, Ziwei Zhang, Peng Cui, Bo Zhang and Leyu Lin. Improving Accuracy and Diversity in Matching of Recommendation with Diversified Preference Network(TBD-2021). (* indicates equal contribution).