node2vec 教程

node2vec 教程

node2vec项目地址:https://gitcode.com/gh_mirrors/no/node2vec

1. 项目目录结构及介绍

在 node2vec 的源代码仓库中，目录结构如下：

node2vec/
├── README.md       # 项目说明文件
├── data/            # 存放数据集的目录
├── scripts/         # 包含辅助脚本的目录
│   ├── preprocess.py    # 数据预处理脚本
│   └── run.sh           # 运行示例脚本
├── src/             # 主要的代码实现
│   ├── graph.cpp      # 图结构的C++实现
│   ├── node2vec.cpp   # node2vec算法的C++实现
│   ├── node2vec.h     # 算法接口头文件
│   ├── util.cpp       # 辅助工具函数
│   └── util.h         # 工具函数头文件
└── python/          # Python接口
    ├── node2vec.py    # Python版本的node2vec实现
    └── __init__.py    # Python包初始化文件

• data/: 存放原始网络数据。
• scripts/: 提供预处理和运行示例的脚本。
• src/: C++实现的node2vec算法核心代码。
• python/: Python封装的接口，方便在Python环境中调用。

2. 项目的启动文件介绍

• run.sh: 这是一个bash脚本，展示了如何使用预处理脚本和C++程序来运行node2vec。首先执行preprocess.py对输入数据进行预处理，然后编译src目录下的C++代码并运行./node2vec来训练模型。你可以根据自己的需求修改此脚本来适应不同的参数设置和数据集。

# 在命令行中运行以下命令以启动示例
bash scripts/run.sh

• scripts/preprocess.py: Python脚本用于将原始数据转换为节点和边的列表格式，这是node2vec算法所需要的输入格式。

# 调用预处理脚本
python scripts/preprocess.py -input input_file.txt -output processed_data.txt

• src/node2vec.cpp: C++实现的主程序，接受预处理后的数据和训练参数来运行node2vec算法。

3. 项目的配置文件介绍

虽然node2vec不使用传统的配置文件，但可以通过命令行参数或者在运行脚本时指定训练过程中的参数。以下是一些关键参数的例子：

• -input: 指定预处理后的输入文件路径。
• -output: 输出特征向量文件的路径。
• -dimensions: 特征向量的维度（默认是128）。
• -walks_per_node: 每个节点执行随机游走的数量（默认是10）。
• -walk_length: 随机游走的步数（默认是80）。
• -window_size: 上下文窗口大小（默认是10）。
• -workers: 并行进程数量（默认是8）。
• -p: 返回参数（控制探索局部结构）（默认是1）。
• -q: 抵达参数（控制探索全局结构）（默认是1）。

例如，在运行C++程序时，可以这样调整参数：

./node2vec -input processed_data.txt -output embeddings.txt -dimensions 64 -walks_per_node 5 -walk_length 40 -window_size 5 -p 0.5 -q 2 -workers 4

这个例子中，节点特征向量的维度设为64，每个节点执行5次随机游走，每次游走长度为40步，窗口大小为5，返回参数p和抵达参数q被用来控制节点邻域的采样策略。

为了在Python环境中使用这些参数，你需要在调用node2vec.py时传递相应的关键字参数，例如：

from node2vec import Node2Vec

model = Node2Vec(graph, dimensions=64, walks_per_node=5, walk_length=40, window_size=5, p=0.5, q=2)
model.fit()

请注意，实际使用前，你可能需要根据你的具体应用场景和数据集大小调整这些参数。

node2vec项目地址:https://gitcode.com/gh_mirrors/no/node2vec