- 环境:requirement.txt
- 训练LightGCN-Social
python main.py --model=LightGCN-Social --dataset=lastfm --decay=1e-4 --lr=0.001 --layer=3 --seed=42 --topks="[10,20]" --recdim=64 --bpr_batch=2048 - 终端预测
python predict.py --user_id ryanlambert --top_k 5python predict.py --user_id 42 --top_k 5 - 网页测试
python app.py
everythinglock/LightGCN-Social
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|