Skip to content

Latest commit

 

History

History
10 lines (10 loc) · 486 Bytes

README.md

File metadata and controls

10 lines (10 loc) · 486 Bytes

Disen-GNN

Paper and code

This is the code for the Paper: Disentangled Graph Neural Networks for Session-based Recommendation. We have implemented our methods in Pytorch.

Usage

The code contains the datasets that has been processed, i.e., diginetica,Nowplaying,Yoo etc. Then you can run the file code\main.py to train the model.
For example: python main.py --dataset diginetica

Requirements

Python 3 ;PyTorch 1.9.1