Jinrui Wei's starred repositories
graphormer-pyg
Microsoft Graphormer (https://arxiv.org/abs/2106.05234) rewritten in Pytorch-Geometric
Graph_Sampling
Graph Sampling is a python package containing various approaches which samples the original graph according to different sample sizes.
SIGIR21-SURGE
Official implementation of SIGIR'2021 paper: "Sequential Recommendation with Graph Neural Networks".
Awesome-Contrastive-Learning-and-Data-Augmentation-RS-Paper-Code
The latest research progress of Contrastive Learning(CL), Data Augmentation(DA) and Self-Supervised Learning(SSL) in Recommender Systems
caser_pytorch
A PyTorch implementation of Convolutional Sequence Embedding Recommendation Model (Caser)
CIKM2020-S3Rec
Code for CIKM2020 "S3-Rec: Self-Supervised Learning for Sequential Recommendation with Mutual Information Maximization"
SASRec.pytorch
PyTorch(1.6+) implementation of https://github.com/kang205/SASRec
BERT4Rec-VAE-Pytorch
Pytorch implementation of BERT4Rec and Netflix VAE.
graphtransformer
Graph Transformer Architecture. Source code for "A Generalization of Transformer Networks to Graphs", DLG-AAAI'21.
neural_collaborative_filtering
Neural Collaborative Filtering
DIB-PEB-Sequential-RS
A tensorflow implementation of the paper "Dynamic Item Block and Prediction Enhancing Block for Sequential Recommendation" in IJCAI 2019
KION_DATASET
This is an official repository of the Kion Movies Recommendation Dataset.
Neural-Attentive-Session-Based-Recommendation-PyTorch
A PyTorch implementation of Neural Attentive Session Based Recommendation (NARM)
Co-Action-Network
Implementation of CAN: Revisiting Feature Co-Action for Click-Through RatePrediction