CogDL TensorFlow:CogDL TensorFlow 源码

qqresponsible79488 23 0 ZIP 2021-04-24 05:04:33

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

用户评论
请输入评论内容
评分:
暂无评论