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

Pytorch implementation of the paper "Enhancing Content Preservation in Text Style Transfer Using Reverse Attention and Conditional Layer Normalization"

Contextualized Perturbation for Textual Adversarial Attack, NAACL 2021

SIR model parameter estimation using a novel algorithm for differentiated uniformization.

Official Code Implementation of the paper : XAI for Transformers: Better Explanations through Conservative Propagation

Real-world Anomaly Detection in Surveillance Videos- pytorch Re-implementation

Gesture recognition on Event Data

Instance-based label smoothing for improving deep neural networks generalization and calibration

Do you like Quick, Draw? Well what if you could train/predict doodles drawn inside Streamlit? Also draws lines, circles and boxes over background images for annotation.

CONditionals for Ordinal Regression and classification in tensorflow

Tensorflow 2.x implementation of Panoramic BlitzNet for object detection and semantic segmentation on indoor panoramic images.

This repository is for Competition for ML_data class

Automated detection of anomalous exoplanet transits in light curve data.

Implementation for paper MLP-Mixer: An all-MLP Architecture for Vision

A user-friendly research and development tool built to standardize RL competency assessment for custom agents and environments.

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

Code for sound field predictions in domains with impedance boundaries. Used for generating results from the paper

Image inpainting using Gaussian Mixture Models

Conceptual 12M is a dataset containing (image-URL, caption) pairs collected for vision-and-language pre-training.

DTCN SMP Challenge - Sequential prediction learning framework and algorithm

Escaping the Gradient Vanishing: Periodic Alternatives of Softmax in Attention Mechanism