Skip to content

Latest commit

 

History

History
38 lines (27 loc) · 1.27 KB

README.md

File metadata and controls

38 lines (27 loc) · 1.27 KB

LightGCN++

Source code and datasets for "Revisiting LightGCN: Unexpected Inflexibility, Inconsistency, and A Remedy Towards Improved Recommendation" (RecSys 2024 Short Paper).

Paper: link

Supplementary document: link

How to Run the Code

The source code of LightGCN++ can be found here.

  • To run LightGCN++ with the specific configuration for each dataset, simply run:
./run.sh
  • For the version supoorting Intel Gaudi devices:
./run_gaudi.sh
  • To run with different $\alpha$, $\beta$, and $\gamma$, run:
python main.py --dataset=[DATASET NAME] --alpha [ALPHA VALUE] --beta [BETA VALUE] --gamma [GAMMA VALUE]

e.g.,
python main.py --dataset="yelp2018" --alpha 0.6 --beta -0.1 --gamma 0.1
  • By default, we recommend using $\alpha=0.6$, $\beta=-0.1$, and $\gamma=0.2$.

Datasets

We used five datasets: LastFM, MovieLens, Gowalla, Yelp, and Amazon. You can find them here.

Acknowledgement

This code is implemented based on the open source LightGCN PyTorch code. This research was supported in part by the NAVER-Intel Co-Lab. The work was conducted by KAIST and reviewed by both NAVER and Intel.