/HERec

Source code for TKDE 2018 "Heterogeneous information network embedding for recommendation"

Primary LanguageTeX

HERec is also available in OpenHINE (https://github.com/BUPT-GAMMA/OpenHINE)

Usage:

  • python cut_data.py

  • python metapathGeneration.py

  • python embeddingGeneration.py

  • python HERec_sl.py / HERec_pl.py / HERec_spl.py

(Note: HERec_sl, HERec_pl, HERec_spl mean HERec with simple linear fusion function, personalized linear fusion and non-linear fusion)

Requirements

Reference

@article{

author = {Chuan Shi, Binbin Hu, Wayne Xin Zhao and Philip S. Yu.},

title = {Heterogeneous Information Network Embedding for Recommendation},

booktitle = {IEEE Transactions on Knowledge and Data Engineering (TKDE)},

year = {2018},

url = {https://arxiv.org/pdf/1711.10730.pdf},

publisher = {IEEE},

keywords = {Heterogeneous information network, Network embedding, Matrix factorization, Recommender system},

}