Heterogeneous Neural Attentive Factorization Machine for Rating Prediction

Tensorflow Implementation of HNAFM

This is our official implementation for the paper: Liang Chen, Yang Liu, Zibin Zheng, and Philip S Yu. 2018. Heterogeneous Neural Attentive Factorization Machine for Rating Prediction. In Proceedings of CIKM'18

If you use the codes, please cite our paper. Thanks!

@inproceedings{chen2018heterogeneous,
  title={Heterogeneous Neural Attentive Factorization Machine for Rating Prediction},
  author={Chen, Liang and Liu, Yang and Zheng, Zibin and Yu, Philip},
  booktitle={Proceedings of the 27th ACM International Conference on Information and Knowledge Management},
  pages={833--842},
  year={2018},
  organization={ACM}
}