Graph Neural Tangent Kernel (GNTK) 项目教程

Graph Neural Tangent Kernel (GNTK) 项目教程

gntkImplementation of Graph Neural Tangent Kernel (NeurIPS 2019)项目地址:https://gitcode.com/gh_mirrors/gn/gntk

1. 项目的目录结构及介绍

gntk/
├── LICENSE
├── README.md
├── dataset.zip
├── gntk.py
├── gram.py
├── run_gram.sh
├── run_search.sh
├── search.py
└── util.py

• LICENSE: 项目的许可证文件，采用MIT许可证。
• README.md: 项目的基本介绍和使用说明。
• dataset.zip: 包含项目使用的数据集。
• gntk.py: 实现Graph Neural Tangent Kernel的主要代码文件。
• gram.py: 用于计算GNTK的gram矩阵。
• run_gram.sh: 运行gram.py的脚本。
• run_search.sh: 运行超参数搜索的脚本。
• search.py: 用于执行超参数搜索的代码文件。
• util.py: 包含项目中使用的各种工具函数。

2. 项目的启动文件介绍

项目的启动文件主要是run_gram.sh和run_search.sh。

• run_gram.sh: 该脚本用于启动计算GNTK的gram矩阵的过程。使用方法如下：

  bash run_gram.sh
  

• run_search.sh: 该脚本用于启动超参数搜索的过程。使用方法如下：

  bash run_search.sh
  

3. 项目的配置文件介绍

项目中没有显式的配置文件，但可以通过命令行参数在运行时进行配置。例如，在运行gram.py时，可以通过以下参数进行配置：

python gram.py --dataset IMDBBINARY --num_mlp_layers 2 --num_layers 2 --scale uniform --jk 1 --out_dir out

• --dataset: 指定使用的数据集。
• --num_mlp_layers: 指定MLP层的数量。
• --num_layers: 指定层的数量。
• --scale: 指定缩放方式。
• --jk: 指定是否使用跳跃连接。
• --out_dir: 指定输出目录。

通过这些参数，可以在运行时灵活配置项目的各种参数。

gntkImplementation of Graph Neural Tangent Kernel (NeurIPS 2019)项目地址:https://gitcode.com/gh_mirrors/gn/gntk