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

GPU-Accelerated Deep Learning Library in Python

This is the official repository of the paper Stocastic bandits with groups of similar arms (NeurIPS 2021). It contains the code that was used to compute the figures and experiments of the paper.

A Comparative Framework for Multimodal Recommender Systems

Outlier Exposure with Confidence Control for Out-of-Distribution Detection

A Home Assistant custom component for Lobe. Lobe is an AI tool that can classify images.

Code for "Adversarial Attack Generation Empowered by Min-Max Optimization", NeurIPS 2021

A project for developing transformer-based models for clinical relation extraction

Bravia core script for python

ResNEsts and DenseNEsts: Block-based DNN Models with Improved Representation Guarantees

Resources for our AAAI 2022 paper: "LOREN: Logic-Regularized Reasoning for Interpretable Fact Verification".

Multiview 3D object detection on MultiviewC dataset through moft3d.

Unofficial PyTorch implementation of SimCLR by Google Brain

This repository contains the code used to quantitatively evaluate counterfactual examples in the associated paper.

Canonical Appearance Transformations

Moiré Attack (MA): A New Potential Risk of Screen Photos [NeurIPS 2021]

A very short and easy implementation of Quantile Regression DQN

Qcover is an open source effort to help exploring combinatorial optimization problems in Noisy Intermediate-scale Quantum(NISQ) processor.

VR Viewport Pose Model for Quantifying and Exploiting Frame Correlations

[ArXiv 2021] One-Shot Generative Domain Adaptation

百度2021年语言与智能技术竞赛机器阅读理解Pytorch版baseline