graph-neural-network topic

List graph-neural-network repositories

scGNN

123
Stars
38
Forks
Watchers

scGNN (single cell graph neural networks) for single cell clustering and imputation using graph neural networks

torch-rgcn

111
Stars
16
Forks
Watchers

A PyTorch implementation of the Relational Graph Convolutional Network (RGCN).

Hyper-SAGNN

80
Stars
20
Forks
Watchers

hypergraph representation learning, graph neural network

graphiler

54
Stars
7
Forks
Watchers

Graphiler is a compiler stack built on top of DGL and TorchScript which compiles GNNs defined using user-defined functions (UDFs) into efficient execution plans.

G3

41
Stars
8
Forks
Watchers

G3: A Programmable GNN Training System on GPU

GEN

59
Stars
11
Forks
59
Watchers

Official Code Repository for the paper "Learning to Extrapolate Knowledge: Transductive Few-shot Out-of-Graph Link Prediction" (NeurIPS 2020)

TIP

24
Stars
15
Forks
Watchers

TIP: Tri-graph Interaction Propagation model for Polypharmacy Side Effect Prediction (GRL@NeurIPS, 2019)

RED-GNN

55
Stars
11
Forks
Watchers

Knowledge Graph Reasoning with Relational Digraph. WebConf 2022

Transformer-M

198
Stars
23
Forks
Watchers

[ICLR 2023] One Transformer Can Understand Both 2D & 3D Molecular Data (official implementation)

ME-GraphAU

155
Stars
39
Forks
Watchers

[IJCAI 2022] Learning Multi-dimensional Edge Feature-based AU Relation Graph for Facial Action Unit Recognition, Pytorch code