pytorch-geometric

There are 8 repositories under pytorch-geometric topic.

THUMNLab / AutoGL
An autoML framework & toolkit for machine learning on graphs.
automl deep-learning graph-neural-networks hyper-parameter-optimization machine-learning neural-architecture-search pytorch pytorch-geometric
Language:Python 1071
graphein
a-r-j / graphein
Protein Graph Library
protein protein-structure deep-learning pytorch graph-neural-networks dgl computational-biology bioinformatics structural-biology protein-data-bank protein-design geometric-deep-learning pytorch-geometric python interactomics interactome rna ppi-networks gene-regulatory-networks drug-discovery
Language:Jupyter Notebook 989
chaitjo / geometric-gnn-dojo
Geometric GNN Dojo provides unified implementations and experiments to explore the design space of Geometric Graph Neural Networks.
geometric-deep-learning graph-neural-networks equivariance equivariant-networks geometric-graphs graph-isomorphism pytorch pytorch-geometric e3nn
Language:Jupyter Notebook 427
lukecavabarrett / pna
Implementation of Principal Neighbourhood Aggregation for Graph Neural Networks in PyTorch, DGL and PyTorch Geometric
graph-machine-learning graph-neural-networks pytorch dgl pytorch-geometric
Language:Python 329
bknyaz / graph_attention_pool
Attention over nodes in Graph Neural Networks using PyTorch (NeurIPS 2019)
graph graph-neural-networks pytorch neurips neurips-2019 attention-mechanism attention pooling pytorch-geometric
Language:Jupyter Notebook 276
qitianwu / DIFFormer
The official implementation for ICLR23 spotlight paper "DIFFormer: Scalable (Graph) Transformers Induced by Energy Constrained Diffusion"
attention diffusion diffusion-equation geometric-deep-learning graph-neural-networks graph-transformer iclr2023 image-classification large-graph node-classification pytorch pytorch-geometric pytorch-geometric-temporal spatial-temporal-forecasting text-classification transformer
Language:Python 273
SGCN
benedekrozemberczki / SGCN
A PyTorch implementation of "Signed Graph Convolutional Network" (ICDM 2018).
gcn graphsage graph-convolution graph-embedding deepwalk node2vec graph-attention network-embedding node-classification signed-network sgcn deep-learning graph-neural-networks neural-network pytorch machine-learning pytorch-geometric transformer gpt2 gpt3
Language:Python 268
qitianwu / NodeFormer
The official implementation of NeurIPS22 spotlight paper "NodeFormer: A Scalable Graph Structure Learning Transformer for Node Classification"
graph-neural-networks graph-structure-learning graph-transformer node-classification pytorch image-classification large-graph neurips-2022 pytorch-geometric relational-learning text-classification
Language:Jupyter Notebook 266
mlabonne / graph-neural-network-course
Free hands-on course about Graph Neural Networks using PyTorch Geometric.
deep-learning graph-attention-networks graph-convolutional-networks graph-neural-networks neural-network pytorch pytorch-geometric
Language:Jupyter Notebook 229
yuyangw / MolCLR
Implementation of MolCLR: "Molecular Contrastive Learning of Representations via Graph Neural Networks" in PyG.
deep-learning graph-neural-networks molecule pytorch pytorch-geometric self-supervised-learning
Language:Python 228
drprojects / DeepViewAgg
[CVPR'22 Best Paper Finalist] Official PyTorch implementation of the method presented in "Learning Multi-View Aggregation In the Wild for Large-Scale 3D Semantic Segmentation"
attention cvpr cvpr2022 deep-learning image kitti-360 multi-view multimodal multimodal-deep-learning point-cloud point-cloud-segmentation pytorch pytorch-geometric s3dis semantic-segmentation torch-points3d
Language:Python 216
divelab / GOOD
GOOD: A Graph Out-of-Distribution Benchmark [NeurIPS 2022 Datasets and Benchmarks]
deep-learning graph-neural-networks out-of-distribution-generalization pytorch graph-ood distribution-shift invariant-learning pytorch-geometric
Language:Python 175
rusty1s / pyg_autoscale
Implementation of "GNNAutoScale: Scalable and Expressive Graph Neural Networks via Historical Embeddings" in PyTorch
graph-neural-networks scalability pytorch pytorch-geometric
Language:Python 157
HannesStark / 3DInfomax
Making self-supervised learning work on molecules by using their 3D geometry to pre-train GNNs. Implemented in DGL and Pytorch Geometric.
graph-representation-learning self-supervised-learning graph molecules 3d pytorch-geometric dgl-graph gnn graph-neural-networks
Language:Python 145
chaitjo / geometric-rna-design
gRNAde: Geometric Deep Learning for RNA Design
geometric-deep-learning graph-neural-networks inverse-design pytorch pytorch-geometric rna-structure biomolecule-design
Language:Jupyter Notebook 117
SherylHYX / pytorch_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.
gnn graph-neural-netowrks signed-networks directed-networks pytorch-geometric pytorch python machine-learning deep-learning networks
Language:Python 114
BorgwardtLab / TOGL
Topological Graph Neural Networks (ICLR 2022)
persistent-homology graph-learning iclr2022 machine-learning graph-classification node-classification pytorch pytorch-geometric
Language:Python 101
hsn
rubenwiersma / hsn
Code for SIGGRAPH paper CNNs on Surfaces using Rotation-Equivariant Features
geometric-deep-learning pytorch-geometric 3d convolutional-neural-networks pytorch siggraph
Language:Python 85
EdisonLeeeee / GreatX
A graph reliability toolbox based on PyTorch and PyTorch Geometric (PyG).
adversarial-attacks graph-convolutional-networks graph-neural-networks pytorch graph-reliability-toolbox distribution-shift inherent-noise pytorch-geometric
Language:Python 81
qitianwu / GraphOOD-GNNSafe
The official implementation for ICLR23 paper "GNNSafe: Energy-based Out-of-Distribution Detection for Graph Neural Networks"
anamoly-detection artificial-intelligence deep-learning distribution-shift geometric-deep-learning graph-neural-networks label-propagation large-graph node-classification out-of-distribution-detection outlier-detection pytorch pytorch-geometric trustworthy-ai out-of-distribution-generalization
Language:Python 68
arangoml / fastgraphml
Given an input graph (ArangoDB or PyG) it generates graph embeddings using Low-Code framework built on top of PyG.
embeddings graph-machine-learning graph-neural-networks machine-learning pytorch-geometric
Language:Python 64
SRA2 / SPELL
Learning Long-Term Spatial-Temporal Graphs for Active Speaker Detection (ECCV 2022)
active-speaker-detection ava-dataset eccv eccv2022 graph-learning pytorch-geometric
Language:Python 63
mukhlishga / gnn-powerflow
Graph Neural Network application in predicting AC Power Flow calculation. Developed with Pytorch Geometric framework. My Master Thesis at Eindhoven University of Technology
machine-learning artificial-intelligence neural-network graph-neural-network power-flow python pytorch pytorch-geometric electrical electrical-engineering
Language:Jupyter Notebook 56
gospodima / Extended-SimGNN
A PyTorch Geometric implementation of SimGNN with some extensions.
pytorch pytorch-geometric graph-neural-networks graph-edit-distance simgnn graph-isomorphism-network deep-learning machine-learning graph-convolutional-networks network-embedding gcn gnn
Language:Python 52
mlvlab / SELAR
Official PyTorch Implementation of "Self-supervised Auxiliary Learning with Meta-paths for Heterogeneous Graphs". NeurIPS 2020.
machine-learning pytorch-geometric graph-neural-networks
Language:Python 51
tk-rusch / gradientgating
Gradient gating (ICLR 2023)
gat gcn graph-neural-networks graphsage heterophily message-passing oversmoothing pytorch-geometric iclr iclr2023 pytorch
Language:Python 50
tk-rusch / GraphCON
GraphCON (ICML 2022)
graph-attention-networks graph-convolutional-networks graph-neural-networks icml icml-2022 oversmoothing pytorch-geometric
Language:Python 49
XiaShan1227 / GNN-Tutorial
B站GNN教程资料
attention-mechanism gat gatv2 gcn gnn graphsage heterogeneous-graph-neural-network link-prediction networkx pytorch pytorch-geometric cluster-gcn
Language:Jupyter Notebook 48
spatiotemporal_datasets
benedekrozemberczki / spatiotemporal_datasets
Spatiotemporal datasets collected for network science, deep learning and general machine learning research.
deep-learning machine-learning gcn gnn time-series time-series-analysis analytics deepwalk node2vec spatial-analysis spatial-data pytorch pytorch-geometric dataset benchmark spatial-data-analysis data-science epidemiology vector-autoregression
46
balcilar / gnn-spectral-expressive-power
Code of "Analyzing the Expressive Power of Graph Neural Networks in a Spectral Perspective" paper published in ICLR2021
gnn graph spectral-analysis expressiveness-analysis pytorch-geometric datasets tensorflow
Language:Python 45
sisaman / LPGNN
Locally Private Graph Neural Networks (ACM CCS 2021)
graph-neural-networks differential-privacy deep-learning pytorch pytorch-geometric graph-deep-learning deep-graph-library
Language:Jupyter Notebook 43
sisaman / GAP
GAP: Differentially Private Graph Neural Networks with Aggregation Perturbation (USENIX Security '23)
deep-learning pytorch differential-privacy graph-neural-networks pytorch-geometric graph-deep-learning privacy-preserving-machine-learning
Language:Jupyter Notebook 40
IntelLabs / GraVi-T
Graph learning framework for long-term video understanding
graph-learning pytorch-geometric video-understanding
Language:Python 37
aravindsankar28 / GraFRank
Code for "Graph Neural Networks for Friend Ranking in Large-scale Social Platforms" (WWW 2021).
graph-neural-network pytorch-geometric dynamic-graphs graph-embedding social-network friend-recommendation
Language:Python 31
pytorch-gnn-research
youyoungjang / pytorch-gnn-research
Various Applications and Descriptions of GNN with Pytorch, Pytorch Geometric.
pytorch graph-neural-networks pytorch-geometric pytorch-biggraph
Language:Jupyter Notebook 30
navid-naderi / WEGL
The implementation code for our paper Wasserstein Embedding for Graph Learning (ICLR 2021).
wasserstein-embedding graph-learning pytorch pytorch-geometric
Language:Jupyter Notebook 29