README.md

March 3, 2025 ยท View on GitHub

PT4Rec: A Universal Prompt-Tuning Framework for Graph Contrastive Learning-Based Recommendations

Wei Xiao, Qifeng Zhou

Paper

image

Requirements

torchnmf
numba
scipy
torch>=1.7.0

Usage

run main.py
(input) PT4Rec

The code repository references SELFRec and CPTPP. Thanks a lot for their work!

If you find this codebase useful, consider citing:

@article{xiao2025pt4rec,
  title={PT4Rec: a universal prompt-tuning framework for graph contrastive learning-based recommendations},
  author={Xiao, Wei and Zhou, Qifeng},
  journal={Machine Learning},
  volume={114},
  number={3},
  pages={1--20},
  year={2025},
  publisher={Springer}
}