Skip to content
/ MPCN Public

MPCN implemented using Pytorch for KDD 2018 Paper "Multi-Pointer Co-Attention Networks for Recommendation"

Notifications You must be signed in to change notification settings

winterant/MPCN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MPCN

Implementation for the paper:
Tay, Yi, Anh Tuan Luu, and Siu Cheung Hui. "Multi-pointer co-attention networks for recommendation." In Proceedings of the 24th ACM SIGKDD International Conference on Knowledge Discovery & Data Mining, pp. 2309-2318. 2018.

Environments

  • python 3.8
  • pytorch 1.70

Dataset

You need to prepare the following documents:

  1. Dataset(data/Digital_Music.json.gz)
    Download from http://deepyeti.ucsd.edu/jianmo/amazon/index.html (Choose Digital Music)

  2. Word Embedding(embedding/glove.6B.50d.txt)
    Download from https://nlp.stanford.edu/projects/glove

Pre-Process

Preprocess origin dataset which is json format to be train.csv,valid.csv and test.csv.

python data_process.py

Running

Train and test:

python main.py --device cuda:0

About

MPCN implemented using Pytorch for KDD 2018 Paper "Multi-Pointer Co-Attention Networks for Recommendation"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages