Skip to content

[NeurIPS 2024] An official source code for paper "End-to-end Learnable Clustering for Intent Learning in Recommendation".

License

Notifications You must be signed in to change notification settings

yueliu1999/ELCRec

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

ELCRec

Requirements

  • Python >= 3.7
  • Pytorch >= 1.2.0
  • tqdm == 4.26.0
  • faiss-gpu==1.7.1

Quick Start

cd ./src
bash ./scripts/run_sports.sh
ELCRec-Sports_and_Outdoors-1
{'Epoch': 0, 'HIT@5': '0.0286', 'NDCG@5': '0.0185', 'HIT@20': '0.0648', 'NDCG@20': '0.0286'}
bash ./scripts/run_beauty.sh
ELCRec-Beauty-1
{'Epoch': 0, 'HIT@5': '0.0529', 'NDCG@5': '0.0355', 'HIT@20': '0.1079', 'NDCG@20': '0.0509'}
bash ./scripts/run_toys.sh
ELCRec-Toys_and_Games-1
{'Epoch': 0, 'HIT@5': '0.0585', 'NDCG@5': '0.0403', 'HIT@20': '0.1138', 'NDCG@20': '0.0560'}
bash ./scripts/run_yelp.sh
ELCRec-Yelp-1
{'Epoch': 0, 'HIT@5': '0.0236', 'NDCG@5': '0.0150', 'HIT@20': '0.0653', 'NDCG@20': '0.0266'}

About

[NeurIPS 2024] An official source code for paper "End-to-end Learnable Clustering for Intent Learning in Recommendation".

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published