librahu / HERec

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

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

• numpy

• scipy

• deepwalk (https://github.com/phanein/deepwalk)

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},

}