/Neural-Collaborative-Filtering

Pytorch implementation of Neural Collaborative Filtering

Primary LanguagePython

Neural-Collaborative-Filtering

Pytorch implementation of Neural Collaborative Filtering, based on https://arxiv.org/abs/1708.05031 user-item pair training is what is used in the paper. The code here allows for experimental training with item-item pairs.

Uses the MovieLens1M dataset