Project Icon

spektral

基于Keras API和TensorFlow 2的Python库,专为图神经网络(GNN)提供简单灵活的框架

Spektral是一个基于Keras API和TensorFlow 2的Python库,专为图神经网络(GNN)提供简单灵活的框架。该库适用于社交网络用户分类、分子性质预测、图生成、节点聚类和链接预测等任务。Spektral包含多种流行的图深度学习层,如GCN、Chebyshev、GraphSAGE、GAT等,并提供丰富的图操作工具。最新版1.0引入了新数据集、新容器、Loader类和transforms模块,简化了数据处理和模型训练。更多信息请参阅官方文档和示例。

Spektral项目介绍

Spektral是一个基于Keras API和TensorFlow 2的Python库,专注于图深度学习。该项目的主要目标是提供一个简单但灵活的框架,用于创建图神经网络(GNNs)。通过Spektral,用户可以进行社交网络用户分类、预测分子属性、通过生成对抗网络(GANs)生成新图、节点聚类、链接预测等一系列基于图数据的任务。

Spektral的主要功能

Spektral实现了一些目前最流行的用于图深度学习的层,包括:

  • 图卷积网络(GCN)
  • 切比雪夫卷积
  • GraphSAGE
  • ARMA卷积
  • 边缘条件卷积(ECC)
  • 图注意力网络(GAT)
  • 神经预测的近似个性化传播(APPNP)
  • 图同构网络(GIN)
  • 扩散卷积

此外,还有许多其他卷积层供使用者选择。同时,Spektral还提供一些池化层,如:

  • MinCut池化
  • DiffPool
  • Top-K池化
  • 自注意力图池化(SAG Pooling)
  • 全局池化
  • 全局门控注意力池化
  • SortPool

这些层和池化方法大大丰富了用户在图深度学习项目中的选择和灵活性。Spektral还包含许多用于表示、操作和转换图的工具,极大地简化了用户的图深度学习项目开发过程。

安装指南

Spektral与Python 3.6及以上版本兼容,并在最新版本的Ubuntu、MacOS和Windows上进行了测试。其他Linux发行版也应当可以运行。

你可以通过PyPi简单地安装Spektral:

pip install spektral

或者从源码安装:

git clone https://github.com/danielegrattarola/spektral.git
cd spektral
python setup.py install  # 或使用 'pip install .'

在Google Colab上安装Spektral也同样简单:

! pip install spektral

Spektral 1.0 的新特性

Spektral 1.0的发布对于该库来说是一个重要的里程碑,带来了许多新功能和改进。以下是一些主要变化和新特性:

  • 新的GraphDataset容器标准化了Spektral处理数据的方式,不影响模型,而是使数据使用更加简便。
  • 新的Loader类能够隐藏创建图批次的复杂性,无论是编写自定义训练循环还是使用Keras的model-dot-fit方法,数据处理变得更简单。
  • 新的transforms模块实现了多种常见的图操作,可以应用于数据集。
  • 新的GeneralConvGeneralGNN类允许用户构建通用模型,不需担心选择哪种层或架构。
  • 新的数据集:QM7和ModelNet10/40,以及OGB数据集的包装器。
  • 库结构和依赖关系的重大清理。
  • 新的示例和教程。

社区贡献

Spektral是一个开源项目,托管在Github上,欢迎各种类型的贡献。如果有想要添加到框架中的有趣的东西,可以随时提出pull request。贡献指南以及功能请求列表均可在GitHub上找到。

通过Spektral,图深度学习变得更加可达并且简单化,为广大开发者开辟了一个广阔的应用天地。

项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手,通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能,支持100+编程语言,与主流编辑器无缝集成,显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台,能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐,Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案,通过智能技术简化视频制作流程。无论是企业宣传还是个人分享,有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持,能够阅读和理解用户上传的文件内容,解析网页信息,并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题,Kimi都能以友好、专业的方式提供帮助。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术,为商家提供一键生成商品图和营销文案的服务,显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台,让商品第一时间被种草。

Project Cover

吐司

探索Tensor.Art平台的独特AI模型,免费访问各种图像生成与AI训练工具,从Stable Diffusion等基础模型开始,轻松实现创新图像生成。体验前沿的AI技术,推动个人和企业的创新发展。

Project Cover

SubCat字幕猫

SubCat字幕猫APP是一款创新的视频播放器,它将改变您观看视频的方式!SubCat结合了先进的人工智能技术,为您提供即时视频字幕翻译,无论是本地视频还是网络流媒体,让您轻松享受各种语言的内容。

Project Cover

美间AI

美间AI创意设计平台,利用前沿AI技术,为设计师和营销人员提供一站式设计解决方案。从智能海报到3D效果图,再到文案生成,美间让创意设计更简单、更高效。

Project Cover

稿定AI

稿定设计 是一个多功能的在线设计和创意平台,提供广泛的设计工具和资源,以满足不同用户的需求。从专业的图形设计师到普通用户,无论是进行图片处理、智能抠图、H5页面制作还是视频剪辑,稿定设计都能提供简单、高效的解决方案。该平台以其用户友好的界面和强大的功能集合,帮助用户轻松实现创意设计。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号