使用 Spektral 在 TensorFlow 和 Keras 中构建图神经网络
机器学习
2020-06-23 v1 机器学习
摘要
在本文中，我们介绍了 Spektral，这是一个用于使用 TensorFlow 和 Keras 应用程序编程接口构建图神经网络的开源 Python 库。Spektral 实现了大量用于图深度学习的方法，包括消息传递和池化算子，以及用于处理图和加载流行基准数据集的工具。该库的目的是提供构建图神经网络的基本组件，并以 Keras 所基于的用户友好和快速原型设计为指导原则。因此，Spektral 既适合初学者，也适合深度学习专家。在这项工作中，我们概述了 Spektral 的功能，并报告了该库实现的方法在节点分类、图分类和图回归场景下的性能。
引用
@article{arxiv.2006.12138,
title = {Graph Neural Networks in TensorFlow and Keras with Spektral},
author = {Daniele Grattarola and Cesare Alippi},
journal= {arXiv preprint arXiv:2006.12138},
year = {2020}
}
备注
ICML 2020 - GRL+ Workshop