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

Support Node2Vec graph sampling algorithm #829

Merged
merged 16 commits into from
Dec 25, 2024

Conversation

AntiTopQuark
Copy link
Contributor

Relate Issue: #716

Use a single thread to generate a Path string, and then use the Word2Vec model to generate a word vector. The word vector will be stored in the Feature part, i.e. NodeInfo[1]

[root@5bcae90fa0cb output]# python3 train_cora.py --method=node2vec_sampling > ./exec.log
/usr/local/lib/python3.6/site-packages/dgl/dataloading/dataloader.py:869: DGLWarning: Dataloader CPU affinity opt is not enabled, consider switching it on (see enable_cpu_affinity() or CPU best practices for DGL [https://docs.dgl.ai/tutorials/cpu/cpu_best_practises.html])
  dgl_warning(f'Dataloader CPU affinity opt is not enabled, consider switching it on '
/usr/local/lib/python3.6/site-packages/dgl/dataloading/dataloader.py:869: DGLWarning: Dataloader CPU affinity opt is not enabled, consider switching it on (see enable_cpu_affinity() or CPU best practices for DGL [https://docs.dgl.ai/tutorials/cpu/cpu_best_practises.html])
  dgl_warning(f'Dataloader CPU affinity opt is not enabled, consider switching it on '
[root@5bcae90fa0cb output]# head -n 5 exec.log
Loading DB state from disk
Loading DB state from disk
Loading DB state from disk
Loading DB state from disk
Loading DB state from disk
[root@5bcae90fa0cb output]# tail -n 5 exec.log
loss:   0.9962489008903503
loss:   0.8431514501571655
loss:   0.8940250277519226
epoch: 1 loss: 0.8940250277519226
The loss value is less than 0.9

@qishipengqsp qishipengqsp merged commit 76fe095 into TuGraph-family:master Dec 25, 2024
5 checks passed
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.

3 participants