Official public repository of paper "Intention Adaptive Graph Neural Network for Category-Aware Session-Based Recommendation"

Last update: Nov 22, 2022

Related tags

Deep Learning IAGNN

Overview

Intention Adaptive Graph Neural Network (IAGNN)

This is the official repository of paper Intention Adaptive Graph Neural Network for Category-Aware Session-Based Recommendation.

If you found this work helpful, please kindly cite the paper as follows:

@article{cui2021intention,
  title={Intention Adaptive Graph Neural Network for Category-aware Session-based Recommendation},
  author={Cui, Chuan and Shen, Qi and Zhu, Shixuan and Pang, Yitong and Zhang, Yiming and Gao, Hanning and Wei, Zhihua},
  journal={arXiv preprint arXiv:2112.15352},
  year={2021}
}

Prerequisite

Install the dependencies by conda

dgl~=0.6.0.post1
ipdb~=0.13.9
numpy~=1.21.2
pretty_errors~=1.2.24
PyMySQL~=1.0.2
scikit_learn~=1.0.2
torch~=1.8.1
TorchSnooper~=0.8
tqdm~=4.62.3

or by pip:

pip install -r requirements.txt

Dataset

GoogleDrive BaiduPan (提取码：2jd1)

Put the downloaded *.pkl files by following this file structure:

|--dataset
   |--diginetica_x
      |--train.pkl
      |--test.pkl
   |--jdata_cd
      |--train.pkl
      |--test.pkl
   |--yc_BT_4
      |--train.pkl
      |--test.pkl
|--IAGNN	# Souce code of this repository
   |--train.py
   |--IAGNN.py
   ...

How to train

# JData
python train.py --lr=0.003 --lr_step=2 --GL=3 --dataset=jdata_cd
# Yoochoose
python train.py --lr=0.001 --lr_step=1 --GL=1 --dataset=yc_BT_4
# Diginetica
python train.py --lr=0.003 --lr_step=1 --GL=2 --dataset=diginetica_x

Official public repository of paper "Intention Adaptive Graph Neural Network for Category-Aware Session-Based Recommendation"

Related tags

Overview

Intention Adaptive Graph Neural Network (IAGNN)

Prerequisite

Dataset

How to train

Owner

Code repo for "FASA: Feature Augmentation and Sampling Adaptation for Long-Tailed Instance Segmentation" (ICCV 2021)

Computer Vision and Pattern Recognition, NUS CS4243, 2022

Turning pixels into virtual points for multimodal 3D object detection.

This is the source code for our ICLR2021 paper: Adaptive Universal Generalized PageRank Graph Neural Network.

Object DGCNN and DETR3D, Our implementations are built on top of MMdetection3D.

Morphable Detector for Object Detection on Demand

BiSeNet based on pytorch

Deep Distributed Control of Port-Hamiltonian Systems

Backdoor Attack through Frequency Domain

Train an imgs.ai model on your own dataset

Code implementation for the paper 'Conditional Gaussian PAC-Bayes'.

Self-supervised Label Augmentation via Input Transformations (ICML 2020)

This is an official implementation of CvT: Introducing Convolutions to Vision Transformers.

3DMV jointly combines RGB color and geometric information to perform 3D semantic segmentation of RGB-D scans.

My course projects for the 2021 Spring Machine Learning course at the National Taiwan University (NTU)

Repository for "Toward Practical Monocular Indoor Depth Estimation" (CVPR 2022)

Python utility to generate filesystem content for Obsidian.

Vrcwatch - Supply the local time to VRChat as Avatar Parameters through OSC

NasirKhusraw - The TSP solved using genetic algorithm and show TSP path overlaid on a map of the Iran provinces & their capitals.

reimpliment of DFANet: Deep Feature Aggregation for Real-Time Semantic Segmentation