6、图嵌入与神经网络：从Node2Vec到图神经网络的探索

图嵌入与神经网络：从Node2Vec到图神经网络的探索

1. Node2Vec的实现

Node2Vec的实现与DeepWalk非常相似，我们可以复用部分代码，通过设置参数 $p = 1$ 和 $q = 1$ ，将DeepWalk作为Node2Vec的特殊情况来实现。以下是使用Zachary’s Karate Club数据集实现Node2Vec的步骤：
1. 安装gensim库 ：为了使用Word2Vec，我们安装版本3.8.0的gensim库。

!pip install -qI gensim==3.8.0

2. 导入所需库 ：

from gensim.models.word2vec import Word2Vec
from sklearn.ensemble import RandomForestClassifier
from sklearn.metrics import accuracy_score

3. 加载数据集 ：

import networkx as nx
G = nx.karate_club_graph()