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

Efficient Sharpness-aware Minimization for Improved Training of Neural Networks

Flow is a computational framework for deep RL and control experiments for traffic microsimulation.

A library for answering questions using data you cannot see

TLoL (Python Module) - League of Legends Deep Learning AI (Research and Development)

JORLDY an open-source Reinforcement Learning (RL) framework provided by KakaoEnterprise

Prometheus exporter for Cisco Unified Computing System (UCS) Manager

This is the repository of our article published on MDPI Entropy "Feature Selection for Recommender Systems with Quantum Computing".

This repository is a series of notebooks that show solutions for the projects at Dataquest.io.

Automatic detection and classification of Covid severity degree in LUS (lung ultrasound) scans

Pip-package for trajectory benchmarking from "Be your own Benchmark: No-Reference Trajectory Metric on Registered Point Clouds", ECMR'21

Implementation of our paper "DMT: Dynamic Mutual Training for Semi-Supervised Learning"

Object-aware Contrastive Learning for Debiased Scene Representation

ICCV2021 Oral SA-ConvONet: Sign-Agnostic Optimization of Convolutional Occupancy Networks

CLDF dataset derived from Robbeets et al.'s "Triangulation Supports Agricultural Spread" from 2021

Python script for performing depth completion from sparse depth and rgb images using the msg_chn_wacv20. model in ONNX

Implementation of ConvMixer-Patches Are All You Need? in TensorFlow and Keras

Turn based roguelike in python

Code for Temporally Abstract Partial Models

Node for thenewboston digital currency network.

Simple tools for logging and visualizing, loading and training