CogDL-TensorFlow
CogDL-TensorFlow :的TensorFlow实现 。 在教授的支持下。
CogDL
CogDL是一种图形表示学习工具包,使研究人员和开发人员可以轻松地训练和比较基线或自定义模型,以进行节点分类,链接预测和其他图形任务。 它提供了许多流行模型的实现,包括:
非GNN基准,例如Deepwalk,LINE,NetMF;
GNN基准,例如GCN,GAT,GraphSAGE。
入门
要求和安装
TensorFlow版本> = 2.0.0
Python版本> = 3.6
pip install -e .
用法
您可以使用python scripts/train.py --task example_task --dataset example_dataset --model example_method在example_data上运行exampl