
CorrectAndSmooth-dgl: 实现DGL中“标签传播与简模型超越图神经网络”方法(ICLR 2021)
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
CorrectAndSmooth-dgl是基于深度图库(DGL)实现的一种创新算法,该算法采用标签传播和简化模型,在不使用复杂图神经网络的情况下超越了现有性能标准,成果发表于ICLR 2021。
DGL CorrectAndSmooth的实现基于论文中的GNN模型。该代码库在Python 3.7环境中运行,并且需要以下软件包版本:dgl 0.6.0.post1、torch 1.7.0 和 ogb 1.3.0。
本示例使用的图形数据集为开放图基准(OGB)。以下是所用数据集的摘要:
- **ogbn-arxiv** 数据集包含节点数:169,343,边的数量:1,166,243,每个节点有128个特征。评价指标是准确性。
- **ogbn-products** 数据集包含节点数量:2,449,029,边的数量:61,859,140,每个节点的特征数为100。评价指标同样是准确性。
使用方法:
训练基本预测模型,并在不同数据集中应用正确的平滑算法(Correct & Smooth),遵循原始超参数设置。
- 对于**ogbn-arxiv** 数据集:`python main.py`
- 使用MLP + C&S的示例命令为:`python main.py --pretrain MLP`
- 使用线性模型+ C&S 的示例命令为: `python main.py --pretrain 线性`
注意,这里的C&S代表Correct & Smooth算法。
全部评论 (0)
还没有任何评论哟~


