GATNE 项目使用教程

1. 项目目录结构及介绍

GATNE/
├── data/
│   ├── amazon/
│   ├── twitter/
│   ├── youtube/
│   └── example/
├── scripts/
│   └── run_example.sh
├── src/
│   ├── main.py
│   ├── main_pytorch.py
│   └── ...
├── .gitignore
├── LICENSE
├── README.md
└── requirements.txt

目录结构介绍

• data/: 存放数据集的目录，包含 Amazon、Twitter、YouTube 和示例数据集。
• scripts/: 包含运行示例的脚本 run_example.sh。
• src/: 项目的主要代码文件，包括 TensorFlow 和 PyTorch 版本的启动文件 main.py 和 main_pytorch.py。
• .gitignore: Git 忽略文件配置。
• LICENSE: 项目许可证文件。
• README.md: 项目介绍和使用说明。
• requirements.txt: 项目依赖库列表。

2. 项目启动文件介绍

main.py

• 功能: 这是 TensorFlow 版本的启动文件，用于训练 GATNE-T 和 GATNE-I 模型。
• 使用方法:

  python src/main.py --input data/example
  

main_pytorch.py

• 功能: 这是 PyTorch 版本的启动文件，用于训练 GATNE-T 和 GATNE-I 模型。
• 使用方法:

  python src/main_pytorch.py --input data/example
  

3. 项目配置文件介绍

config.yaml

• 位置: 配置文件通常位于项目的根目录或 src/ 目录下。
• 内容:
  • use_cuda: 是否使用 GPU 进行训练。
  • data_path: 数据集的存储路径。
  • lr: 学习率。
  • neg_num: 负样本数量。
  • num_walks: 每个节点开始的随机游走次数。
  • walk_length: 随机游走的长度。

配置文件示例

use_cuda: True
data_path: "/data/youtube"
lr: 0.001
neg_num: 5
num_walks: 10
walk_length: 80

修改配置文件

在训练模型之前，可以通过修改 config.yaml 文件中的参数来调整模型的训练行为。例如，如果希望使用 GPU 进行训练，可以将 use_cuda 设置为 True。

use_cuda: True

通过以上步骤，您可以顺利地启动和配置 GATNE 项目，并根据需要调整模型的训练参数。