FangShuheng / HIRE

A PyTorch implementation of HIRE, as described in the paper: Shuheng Fang, Kangfei Zhao, Yu Rong, Zhixun Li, Jeffery Xu Yu. [All-in-One: Heterogeneous Interaction Modeling for Cold-Start Rating Prediction]

All-in-One: Heterogeneous Interaction Modeling for Cold-Start Rating Prediction

A PyTorch implementation of HIRE, as described in the paper: Shuheng Fang, Kangfei Zhao, Yu Rong, Zhixun Li, Jeffery Xu Yu. [All-in-One: Heterogeneous Interaction Modeling for Cold-Start Rating Prediction]

Requirements

python 3.8
numpy
scipy
scikit-learn
torch 1.7.1

Quick Start

Running Movielens

python main.py    \
       --data_set movielens     \

Contact

Open an issue or send email to [e-mail] if you have any problem

Cite Us