Graph Machine Learning Python Packages

ogb

Benchmark datasets, data loaders, and evaluators for graph machine learning

graphdatascience

A Python client for the Neo4j Graph Data Science (GDS) library

stellargraph

StellarGraph - Machine Learning on Graphs

graph-embeddings

Graph and Nodes embeddings for downstream tasks

gdsclient

A Python client for the Neo4j Graph Data Science (GDS) library

hugegraph-python

A Python SDK for Apache HugeGraph

quantum-evolution-kernel

A Graph Machine Learning library using Quantum Computing

msd-dataset

MSD: A Benchmark Dataset for Floor Plan Generation of Building Complexes

fastgraphml

Given an input graph (ArangoDB or PyG) it generates graph embeddings using Low-Code framework built on top of PyG.

hugegraph-llm

The integration of HugeGraph with AI/LLM & GraphRAG

stellargraph-mvisani

StellarGraph - Machine Learning on Graphs

gigl

TODO

tigerlily

TigerLily: Finding drug interactions in silico with the Graph.