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

The integration of HugeGraph with AI/LLM & GraphRAG

msd-dataset

[ECCV 2024] MSD: A Benchmark Dataset for Floor Plan of Building Complexes

quantum-evolution-kernel

A Python library designed for the machine learning community to help users design quantum-driven similarity metrics for graphs and to use them inside kernel-based machine learning algorithms for graph data.ide the right environment to explore new ideas - both in terms of methodologies and data domain - while always interacting with a simple and intuitive QPU interface.

hugegraph-llm

The integration of HugeGraph with AI/LLM & GraphRAG

fastgraphml

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

gigl

TODO

stellargraph-mvisani

StellarGraph - Machine Learning on Graphs

tigerlily

TigerLily: Finding drug interactions in silico with the Graph.