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

CAMPARI: Camera-Aware Decomposed Generative Neural Radiance Fields

[ACL 2022] LinkBERT: A Knowledgeable Language Model 😎 Pretrained with Document Links

Causal-Adversarial-Instruments - PyTorch Implementation for Developing Library of Investigating Adversarial Examples on A Causal View by Instruments

Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.

Implementation of Kaneko et al.'s MaskCycleGAN-VC model for non-parallel voice conversion.

Code for Universal Semi-Supervised Semantic Segmentation models paper accepted in ICCV 2019

Annotate with anyone, anywhere.

A Python Package for Convex Regression and Frontier Estimation

PyTorch Code for NeurIPS 2021 paper Anti-Backdoor Learning: Training Clean Models on Poisoned Data.

Automatically creates genre collections for your Plex media

the official code for ICRA 2021 Paper: "Multimodal Scale Consistency and Awareness for Monocular Self-Supervised Depth Estimation"

CUAD

Efficient Multi Collection Style Transfer Using GAN

Stratified Transformer for 3D Point Cloud Segmentation (CVPR 2022)

Official implementation of "Open-set Label Noise Can Improve Robustness Against Inherent Label Noise" (NeurIPS 2021)

Accelerated deep learning R&D

MetaDrive: Composing Diverse Scenarios for Generalizable Reinforcement Learning

This repo contains the official implementations of EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasis

A pytorch reproduction of { Co-occurrence Feature Learning from Skeleton Data for Action Recognition and Detection with Hierarchical Aggregation }.

Multi-Scale Aligned Distillation for Low-Resolution Detection (CVPR2021)