/personalized_embedding

Primary LanguagePython

This is the source code of the paper: NPE: Neural Personalized Embedding for Collaborative Filtering (IJCAI 2018).

Run

NPE: python test.py --model EMB-CF Matrix factorization: python test.py --model mf

Data

Two column-csv file. One column for user_id, the other for the item_id.

user_id item_id 0 0 0 5 1 3 1 2