Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fix LightGCN #33

Merged
merged 1 commit into from
Sep 29, 2024
Merged

Fix LightGCN #33

merged 1 commit into from
Sep 29, 2024

Conversation

zechengz
Copy link
Member

@zechengz zechengz commented Sep 28, 2024

Fix the bug that dst node ids should add num_src_nodes instead of num_dst_nodes

python3 light_gcn.py --dataset rel-hm --task user-item-purchase --val_loss
Best val metrics: {'link_prediction_precision': np.float64(0.0030048050061459376), 'link_prediction_recall': np.float64(0.010978732296913969), 'link_prediction_map': np.float64(0.003177537802166651)}
Best test metrics: {'link_prediction_precision': np.float64(0.0025231840025418005), 'link_prediction_recall': np.float64(0.009660929381309786), 'link_prediction_map': np.float64(0.002673721168489858)}

@zechengz zechengz requested a review from rusty1s September 28, 2024 22:42
@zechengz zechengz self-assigned this Sep 28, 2024
@zechengz zechengz merged commit 2408e78 into master Sep 29, 2024
2 checks passed
@zechengz zechengz deleted the zecheng_fix_lightgcn branch September 29, 2024 06:31
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants