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

To Design and Implement Logistic Regression to Classify Between Benign and Malignant Cancer Types

pytorch implementation of trDesign

A fast and easy to use, moddable, Python based Minecraft server!

An Open-Source Package for Information Retrieval.

《K-Adapter: Infusing Knowledge into Pre-Trained Models with Adapters》(2020)

Benchmark tools for Compressive LiDAR-to-map registration

Simultaneous Detection and Segmentation

Self-supervised learning optimally robust representations for domain generalization.

A programming language written with python

Official source code of paper 'IterMVS: Iterative Probability Estimation for Efficient Multi-View Stereo'

Pytorch implementation of TailCalibX : Feature Generation for Long-tail Classification

A deep learning CNN model to identify and classify and check if a person is wearing a mask or not.

Python library for tracking human heads with FLAME (a 3D morphable head model)

Papers about explainability of GNNs

Monify: an Expense tracker Program implemented in a Graphical User Interface that allows users to keep track of their expenses

Deep Networks with Recurrent Layer Aggregation

MoveNetを用いたPythonでの姿勢推定のデモ

TensorFlow implementation of "Attention is all you need (Transformer)"

It helps user to learn Pick-up lines and share if he has a better one

Monk is a low code Deep Learning tool and a unified wrapper for Computer Vision.