文件名称：jraph:Jax中的图神经网络库
文件大小：104KB
文件格式：ZIP
更新时间：2024-03-30 21:38:41
machine-learning deep-learning jax graph-neural-networks Python
Jraph-jax中的图神经网络库。 | Jraph（发音为“长颈鹿”）是一个轻量级的库，用于在jax中使用图神经网络。 它提供了图的数据结构，一组使用图的工具以及可插拔图神经网络模型的“动物园”。 安装 可以使用以下命令直接从github安装Jraph： pip install git+git://github.com/deepmind/jraph.git 概述 Jraph旨在提供用于在jax中处理图形的实用程序，但并未规定编写或开发图形神经网络的方法。 graph.py提供了一个轻量级的数据结构GraphsTuple ，用于处理图形。 utils.py提供了在jax中使用GraphsTuples 。 批处理GraphsTuples数据集的实用工具。 通过填充和遮罩支持可变形状图形的jit编译的实用程序。 用于定义输入分区损失的实用程序。 models.py提供了不同
【文件预览】：
jraph-master
----readthedocs.yml(244B)
----images()
--------logo.png(29KB)
----requirements.txt(41B)
----CONTRIBUTING.md(1KB)
----LICENSE(11KB)
----setup.py(3KB)
----README.md(6KB)
----docs()
--------conf.py(5KB)
--------requirements.txt(142B)
--------index.rst(6KB)
--------Makefile(634B)
--------api.rst(1KB)
----jraph()
--------__init__.py(4KB)
--------examples()
--------ogb_examples()
--------_src()