IRGNN_TNNLS_2021

This repo is a pytorch implementation of the paper "Item Relationship Graph Neural Networks for E-commerce", built upon PyG library.

Dependecies

• Python3.7
• PyTorch
• PyG
• networkx
• pandas
• gensim

Experiment Data

• Amazon Review Data
• To process a dataset from raw files
  • Please get the following files from Amazon Review Data and put them at the raw/ directory.

    meta_Video_Games.json.gz
    reviews_Video_Games.json.gz

  • Obtain node features from reviews by gensim.models.doc2vec, and put the .d2v file at raw/.

Experiment

• Before running, please modify the corresponding Amazon data category in amazon_data_loader.py.

  python run_irgnn.py