Описание тега pytorch-geometric

Pytorch Geometric(PyG) — это библиотека для графовых нейронных сетей (GNN), основанная на PyTorch. Он содержит различные методы написания и обучения GNN на графиках из различных опубликованных статей. Он поддерживает мини-пакетные загрузчики для работы на графических процессорах.

Функции

• Поддерживает различные GNN, см. шпаргалку или список реализованных моделей GNN, многие из которых содержат небольшие примеры, такие как Graph Attention Networks (GAT), Graph Convolutional Networks.
• Поддерживает различные графы: разреженные графы, графы с весами ребер, графы с атрибутами ребер, двудольные графы и т. д.
• Содержит самые известные наборы данных , такие как Cora или Zachary's karate club, а также поддерживается многими другими, такими как Open Graph Benchmark .
• Поддерживает GraphGymAPI для разработки и оценки GNN

Видео и учебник по блокнотам Colab