Gnn Python Packages

torch-geometric-temporal

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)

10K 3K 406

tensorflow-gnn

TensorFlow GNN is a library to build Graph Neural Networks on the TensorFlow platform.

3K 2K 198

torch-geometric-signed-directed

PyTorch Geometric Signed Directed is a signed/directed graph neural network extension library for PyTorch Geometric. The paper is accepted by LoG 2023.

2K 147 21

city2graph

Transform geospatial relations into graphs for Graph Neural Networks and network analysis

2K 1K 128

tf-geometric

Efficient and Friendly Graph Neural Network Library for TensorFlow 1.x and 2.x.

2K 508 92

torch-spatiotemporal

A PyTorch library for spatiotemporal data processing

1K 379 36

causing

Causing: CAUsal INterpretation using Graphs

991 62 9

fennlp

1. Use BERT, ALBERT and GPT2 as tensorflow2.0's layer. 2. Implement GCN, GAN, GIN and GraphSAGE based on message passing.

990 336 66

hugegraph-python

The integration of HugeGraph with AI/LLM & GraphRAG

742 131 77

unravelsports

The unravelsports package aims to aid researchers, analysts and enthusiasts by providing intermediary steps in the complex process of turning raw sports data into meaningful information and actionable insights.

711 240 32

solitairenet

Solitaire deep learning library.

691 0 0

spiga

SPIGA: Shape Preserving Facial Landmarks with Graph Attention Networks.

649 343 42

lamda-ssl

30 Semi-Supervised Learning Algorithms

598 210 16

pandadock

Molecular docking with SE(3)-equivariant GNN scoring - achieves R=0.88 on PDBbind

575 96 18

mlx-cluster

High-performance graph algorithms optimized for Apple's MLX framework. Features random walks, biased random walks (Node2Vec), and neighbor sampling

517 6 0

ptgnn

A PyTorch Graph Neural Network Library

476 375 39

modssc

ModSSC: A Modular Framework for Semi Supervised Classification

476 32 6

graph-transformer

An unofficial implementation of Graph Transformer (Masked Label Prediction: Unified Message Passing Model for Semi-Supervised Classification) - IJCAI 2021

454 35 6