神奇的PyG：快速实现Node2vector

公众号：异度侵入
“我最早接触数据挖掘，还没学会调参，只会用树模型时，比赛时便在群里听别人讨论图神经网络，谈起Deepwalk，Node2Vector，听起来唬人的算法，觉得这样的算法离我很远，好想使用啊，然后现在我遇到了PyG，一切都不一样了”

01 PyG

什么是PyG?PyG，全称Pytorch Geometric，听见这个名字就知道它是pytorch的生态圈的，囊括了许多图算法。比如Node2vector，GCN，GraphSage。https://pytorch-geometric.readthedocs.io/en/latest/。那先来安装一下吧。

pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.8.0+cu111.html
pip install torch-geometric

支持的pytorch版本1.4——1.8。(https://pytorch-geometric.com/whl/）

02 Node2vector

说起Node