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

Implementation of Bottleneck Transformer in Pytorch

natural image generation using ConvNets

Code for ICCV2021 paper PARE: Part Attention Regressor for 3D Human Body Estimation

This demo showcase the use of onnxruntime-rs with a GPU on CUDA 11 to run Bert in a data pipeline with Rust.

Python scripts form performing stereo depth estimation using the CoEx model in ONNX.

This is a simple framework to make object detection dataset very quickly

Vignette is a face tracking software for characters using osu!framework.

Wanli Li and Tieyun Qian: Exploit a Multi-head Reference Graph for Semi-supervised Relation Extraction, IJCNN 2021

List some popular DeepFake models e.g. DeepFake, FaceSwap-MarekKowal, IPGAN, FaceShifter, FaceSwap-Nirkin, FSGAN, SimSwap, CihaNet, etc.

code for our ECCV-2020 paper: Self-supervised Video Representation Learning by Pace Prediction

Official Pytorch implementation of 'RoI Tanh-polar Transformer Network for Face Parsing in the Wild.'

[peer review] An Arbitrary Scale Super-Resolution Approach for 3D MR Images using Implicit Neural Representation

Spatial Action Maps for Mobile Manipulation (RSS 2020)

HiFi-GAN: Generative Adversarial Networks for Efficient and High Fidelity Speech Synthesis

A PyTorch implementation for Unsupervised Domain Adaptation by Backpropagation(DANN), support Office-31 and Office-Home dataset

The code for our paper "AutoSF: Searching Scoring Functions for Knowledge Graph Embedding"

ObjectDrawer-ToolBox: a graphical image annotation tool to generate ground plane masks for a 3D object reconstruction system

Alphabetical Letter Recognition

Experiments and examples converting Transformers to ONNX

Histology images query (unsupervised)