跳转到主要内容
ToolPotion
AI 顾问
提交(免费)

关系图卷积网络

AI GitHub 仓库AI 模型与大语言模型开源 · Linux, macOS, Windows

该项目提供了关系图卷积网络 (R-GCNs) 的 Keras 实现。它专为有向关系图上的半监督节点分类而设计。该代码有助于复现论文“使用图卷积网络对关系数据进行建模”中的实体分类结果。

查看仓库
分享
查看仓库
分享

tkipf/relational-gcn 存储库提供了关系图卷积网络 (R-GCNs) 的 Keras 实现。该工具专门用于有向关系图上的半监督节点分类任务。它为旨在复现或扩展论文“使用图卷积网络对关系数据进行建模”(2017)中提出的研究成果的研究人员和实践者提供了一个宝贵的资源。

阅读更多

该实现侧重于关系数据建模的实体分类方面。用户可以利用此代码在各种数据集上试验 R-GCN,包括 AIFB、MUTAG、BGS 和 AM,具体细节请参阅使用说明。该项目强调了特定依赖版本的重要性,推荐使用 Keras 1.2.1 和 Theano 0.9.0,因为较新版本的 Keras 可能会破坏 Theano 的稀疏矩阵 API。虽然 TensorFlow 可以用作后端,但用户应注意稀疏矩阵大小可能存在的限制。

为确保实验结果的准确复现,该存储库提供了数据集准备和模型训练的清晰说明。它还包括有关配置 Keras 后端为 Theano 和强制执行 CPU 运行的指导,由于某些实验可能存在 GPU 内存限制,因此推荐使用 CPU 运行。该项目在 MIT 许可下发布,鼓励开放访问和修改。相关论文 [1] 提供了对 R-GCN 模型及其在关系数据建模中的应用的更深入的理论理解。

此实现对于处理实体之间关系至关重要的图结构数据的用户特别有用。它能够应用先进的深度学习技术来理解和分类复杂关系网络中的节点。该代码在 GitHub 上公开可用,促进了图神经网络社区内的协作和进一步发展。

  • 类别AI 模型与大语言模型
  • 定价模式开源
  • 部署自托管
  • 平台Linux, macOS, Windows

  • 基于 Keras 的实现

  • 关系图卷积网络 (R-GCNs)

  • 半监督节点分类

  • 支持有向关系图

  • 用于实体分类复现的代码

  • 包含数据集准备脚本

  • 提供模型训练脚本

  • 依赖管理指南

  • 强制 CPU 运行选项

  • MIT 许可

  1. 安装依赖:确保 Keras 和 Theano 版本兼容后,运行 'python setup.py install'。

  2. 配置 Keras 后端:创建 '~/.keras/keras.json' 文件,将 'backend' 设置为 'theano'。

  3. 准备数据集:对特定数据集(例如 aifb、mutag)运行 'python prepare_dataset.py -d <dataset_name>'。

  4. 强制 CPU 运行:在训练命令前加上 'CUDA_VISIBLE_DEVICES='。

  5. 训练模型:运行 'python train.py -d <dataset_name> --bases <num> --hidden <num> --l2norm <value> --testing'。

  6. 复现论文结果:遵循 AIFB、MUTAG、BGS 和 AM 的数据集特定说明。

  7. 适配新任务:修改脚本以适应自定义数据集或关系图任务的变体。

  • 节点分类
  • 实体分类
  • 关系数据建模
  • 图神经网络研究
  • 基于图的机器学习

关系图卷积网络 评价

加载中...

与 关系图卷积网络 类似的热门AI工具

CompGCN

AI GitHub 仓库

CompGCN 是一个基于组合的多关系图卷积网络的 PyTorch 实现。它能够实现先进的图表示学习,用于知识图谱链接预测等任务,提供灵活的组合操作和与各种评分函数的兼容性。

AI 模型与大语言模型
查看工具

AI GitHub 仓库

基于组合的多关系图卷积网络的 PyTorch 实现。它能够实现先进的图表示学习,用于知识图谱链接预测等任务,提供灵活的组合操作和与各种评分函数的兼容性。CompGCN 是一个基于组合的多关系图卷积网络的 PyTorch 实现。它能够实现先进的图表示学习,用于知识图谱链接预测等任务,提供灵活的组合操作和与各种评分函数的兼容性。

AI 模型与大语言模型
查看 AI GitHub 仓库 Top 250

简化图卷积网络

AI GitHub 仓库

此存储库提供了“简化图卷积网络”论文的官方实现。它提供了一个简化的图卷积网络 (GCN) 模型,该模型消除了非线性并折叠了权重矩阵,从而得到了一个线性模型,该模型在显著缩短的训练时间内实现了具有竞争力的性能。

AI 模型与大语言模型
查看工具

AI GitHub 仓库

此存储库提供了“简化图卷积网络”论文的官方实现。它提供了一个简化的图卷积网络 (GCN) 模型,该模型消除了非线性并折叠了权重矩阵,从而得到了一个线性模型,该模型在显著缩短的训练时间内实现了具有竞争力的性能。

AI 模型与大语言模型
查看 AI GitHub 仓库 Top 250

图卷积网络

AI GitHub 仓库

图卷积网络(GCN)是一种专门用于处理图结构数据的神经网络。它将卷积神经网络(CNN)的概念推广到图结构数据,从而在半监督学习和图嵌入等领域实现强大的应用。该模型为分析复杂网络数据提供了一种新颖的方法。

其他 AI 工具
查看工具

AI GitHub 仓库

图卷积网络(GCN)是一种专门用于处理图结构数据的神经网络。它将卷积神经网络(CNN)的概念推广到图结构数据,从而在半监督学习和图嵌入等领域实现强大的应用。该模型为分析复杂网络数据提供了一种新颖的方法。

其他 AI 工具
查看 AI GitHub 仓库 Top 250

GCNII

AI GitHub 仓库

该存储库提供了“简单深度图卷积网络”(GCNII)模型的 PyTorch 实现。它包含在基准图数据集上复现半监督和全监督结果的代码,以及一个 PyTorch Geometric 参考实现。

机器学习平台
查看工具

AI GitHub 仓库

该存储库提供了“简单深度图卷积网络”(GCNII)模型的 PyTorch 实现。它包含在基准图数据集上复现半监督和全监督结果的代码,以及一个 PyTorch Geometric 参考实现。

机器学习平台
查看 AI GitHub 仓库 Top 250

SEAL 链接预测

AI GitHub 仓库

SEAL 是一个新颖的链接预测框架,它将问题转化为子图分类任务。它提取 h-hop 包络子图和节点信息矩阵,并将它们输入图神经网络。SEAL 同时从图结构和潜在/显式节点特征中学习,以实现准确的链接预测。

AI 模型与大语言模型
查看工具

AI GitHub 仓库

新颖的链接预测框架,它将问题转化为子图分类任务。它提取 h-hop 包络子图和节点信息矩阵,并将它们输入图神经网络。SEAL 同时从图结构和潜在/显式节点特征中学习,以实现准确的链接预测。SEAL 是一个新颖的链接预测框架,它将问题转化为子图分类任务。它提取 h-hop 包络子图和节点信息矩阵,并将它们输入图神经网络。SEAL 同时从图结构和潜在/显式节点特征中学习,以实现准确的链接预测。

AI 模型与大语言模型
查看 AI GitHub 仓库 Top 250

LightGCN

AI GitHub 仓库

LightGCN 是一种简化的图卷积网络,专为推荐系统设计。它专注于协同过滤的核心邻域聚合组件,提供了一种从交互数据中学习用户和物品表示的高效方法。

AI 模型与大语言模型
查看工具

AI GitHub 仓库

简化的图卷积网络,专为推荐系统设计。它专注于协同过滤的核心邻域聚合组件,提供了一种从交互数据中学习用户和物品表示的高效方法。LightGCN 是一种简化的图卷积网络,专为推荐系统设计。它专注于协同过滤的核心邻域聚合组件,提供了一种从交互数据中学习用户和物品表示的高效方法。

AI 模型与大语言模型
查看 AI GitHub 仓库 Top 250

Temporal Graph Networks

AI GitHub 仓库

Temporal Graph Networks (TGNs) 提供了一个通用的、高效的动态图深度学习框架。这个基于 Python 的库利用新颖的内存模块和基于图的运算符,在计算效率和演进图结构上的预测准确性方面优于现有方法。

AI 模型与大语言模型
查看工具

AI GitHub 仓库

提供了一个通用的、高效的动态图深度学习框架。这个基于 Python 的库利用新颖的内存模块和基于图的运算符,在计算效率和演进图结构上的预测准确性方面优于现有方法。Temporal Graph Networks (TGNs) 提供了一个通用的、高效的动态图深度学习框架。这个基于 Python 的库利用新颖的内存模块和基于图的运算符,在计算效率和演进图结构上的预测准确性方面优于现有方法。

AI 模型与大语言模型
查看 AI GitHub 仓库 Top 250

DeepGCNs

AI GitHub 仓库

该 GitHub 仓库提供了 DeepGCNs、DeeperGCN 和 GNN1000 的 PyTorch 实现。它通过借鉴 CNN 的残差连接和空洞卷积等概念,实现了非常深层的图卷积网络训练,从而促进了图神经网络的研究。

机器学习平台
查看工具

AI GitHub 仓库

该 GitHub 仓库提供了 DeepGCNs、DeeperGCN 和 GNN1000 的 PyTorch 实现。它通过借鉴 CNN 的残差连接和空洞卷积等概念,实现了非常深层的图卷积网络训练,从而促进了图神经网络的研究。

机器学习平台
查看 AI GitHub 仓库 Top 250
1/8