/NCF

A pytorch implementation of He et al. "Neural Collaborative Filtering" at WWW'17

Primary LanguagePython

NCF

A pytorch GPU implementation of He et al. "Neural Collaborative Filtering" at WWW'17

Note that I use the two sub datasets provided by Xiangnan's repo.

I randomly utilized a factor number 32, MLP layers 3, epochs is 20, and posted the results in the original paper and this implementation here. I employed the exactly same settings with Xiangnan, including batch_size, learning rate, and all the initialization methods in Xiangnan's keras repo. From the results I observed, this repo can replicate the performance of the original NCF. Xiangnan's keras repo:

Models MovieLens HR@10 MovieLens NDCG@10 Pinterest HR@10 Pinterest NDCG@10
MLP 0.692 0.425 0.868 0.542
GMF - - - -
NeuMF (without pre-training) 0.701 0.425 0.870 0.549
NeuMF (with pre-training) 0.726 0.445 0.879 0.555

This pytorch code:

Models MovieLens HR@10 MovieLens NDCG@10 Pinterest HR@10 Pinterest NDCG@10
MLP 0.691 0.416 0.866 0.537
GMF 0.708 0.429 0.867 0.546
NeuMF (without pre-training) 0.701 0.424 0.867 0.544
NeuMF (with pre-training) 0.720 0.439 0.879 0.555

The requirements are as follows:

* python==3.6
* pandas==0.24.2
* numpy==1.16.2
* pytorch==1.0.1
* gensim==3.7.1
* tensorboardX==1.6 (mainly useful when you want to visulize the loss, see https://github.com/lanpa/tensorboard-pytorch)

Example to run:

python main.py --batch_size=256 --lr=0.001 --factor_num=16