Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
Guangyu Zhou eaa51ac4dc | 3 months ago | |
---|---|---|
.. | ||
deepwalk_trainer.py | 3 months ago | |
readme.md | 1 year ago |
Paper link: https://arxiv.org/abs/1403.6652
Author's code repo (in Tensorflow):
Popular dgl implementation:
https://github.com/dmlc/dgl/tree/master/examples/pytorch/ogb/deepwalk.
Dataset | # Nodes | # Edges | # Classes |
---|---|---|---|
Cora | 2,708 | 10,556 | 7 |
Citeseer | 3,327 | 9,228 | 6 |
Pubmed | 19,717 | 88,651 | 3 |
Refer to Planetoid.
TL_BACKEND="tensorflow" python deepwalk_trainer.py --lr 0.05 --embedding_dim 128 --walk_length 20 --window_size 10 --num_walks 10 --n_epoch 100 --dataset cora
TL_BACKEND="tensorflow" python deepwalk_trainer.py --lr 0.05 --embedding_dim 128 --walk_length 20 --window_size 10 --num_walks 10 --n_epoch 100 --dataset citeseer
TL_BACKEND="tensorflow" python deepwalk_trainer.py --lr 0.05 --embedding_dim 64 --walk_length 10 --window_size 5 --num_walks 10 --n_epoch 100 --dataset pubmed
TL_BACKEND="torch" python deepwalk_trainer.py --lr 0.05 --embedding_dim 128 --walk_length 20 --window_size 10 --num_walks 10 --n_epoch 100 --dataset cora
TL_BACKEND="torch" python deepwalk_trainer.py --lr 0.05 --embedding_dim 128 --walk_length 20 --window_size 10 --num_walks 10 --n_epoch 100 --dataset citeseer
TL_BACKEND="torch" python deepwalk_trainer.py --lr 0.05 --embedding_dim 64 --walk_length 10 --window_size 5 --num_walks 10 --n_epoch 100 --dataset pubmed
TL_BACKEND="paddle" python deepwalk_trainer.py --lr 0.05 --embedding_dim 128 --walk_length 20 --window_size 10 --num_walks 10 --n_epoch 100 --dataset cora
TL_BACKEND="paddle" python deepwalk_trainer.py --lr 0.05 --embedding_dim 128 --walk_length 20 --window_size 10 --num_walks 10 --n_epoch 100 --dataset citeseer
TL_BACKEND="paddle" python deepwalk_trainer.py --lr 0.05 --embedding_dim 64 --walk_length 10 --window_size 5 --num_walks 10 --n_epoch 100 --dataset pubmed
Dataset | GCN(report) | Our(pd) | Our(torch) | Our(tf) |
---|---|---|---|---|
cora | 67.2 | 71.20(±1.13) | 70.84(±1.51) | 70.64(±0.96) |
citeseer | 43.2 | 46.72(±1.09) | 47.13(±1.41) | 48.04(±1.96) |
pubmed | 65.3 | 62.61(±2.41) | 61.74(±2.32) | 62.74(±2.43) |
No Description
Python C++ Cuda Markdown Text
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》