Skip to main content

Python package for graph-based clustering and semi-supervised learning

Project description

Graph-based Clustering and Semi-Supervised Learning


This python package is devoted to efficient implementations of modern graph-based learning algorithms for semi-supervised learning, active learning, and clustering. The package implements many popular datasets (currently MNIST, FashionMNIST, and CIFAR-10) in a way that makes it simple for users to test out new algorithms and rapidly compare against existing methods. Full documentation is available, including detailed example scripts.

This package also reproduces experiments from the paper

J. Calder, B. Cook, M. Thorpe, D. Slepcev. Poisson Learning: Graph Based Semi-Supervised Learning at Very Low Label Rates., Proceedings of the 37th International Conference on Machine Learning, PMLR 119:1306-1316, 2020.


Install with

pip install graphlearning

Required packages will be installed automatically, and include numpy, scipy, sklearn, and matplotlib. Some features in the package rely on other packages, including annoy for approximate nearest neighbor searches, and torch for GPU acceleration. You will have to install these manually, if needed, with

pip install annoy torch

It can be difficult to install annoy, depending on your operating system.

To install the most recent version of GraphLearning from the github source, which is updated more frequently, run

git clone
cd GraphLearning
pip install .

If you prefer to use ssh swap the first line with

git clone

Documentation and Examples

Full documentation for the package is available here. The documentation includes examples of how to use the package. All example scripts linked from the documentation can be found in the examples folder.

Older versions of GraphLearning

This repository hosts the current version of the package, which is numbered >=1.0.0. This version is not backwards compatible with earlier versions of the package. The old version is archived here and can be installed with

pip install graphlearning==0.0.3

To make sure you will load the old version when running import graphlearning, it may be necessary to uninstall all existing versions pip uninstall graphlearning before running the installation command above.


If you use this package in your research, please cite the package with the bibtex entry below.

  author       = {Jeff Calder},
  title        = {GraphLearning Python Package},
  month        = jan,
  year         = 2022,
  publisher    = {Zenodo},
  doi          = {10.5281/zenodo.5850940},
  url          = {}

Contact and questions

Email with any questions or comments.


Several people have contributed to the development of this software:

  1. Mauricio Rios Flores (Machine Learning Researcher, Amazon)
  2. Brendan Cook (PhD Candidate in Mathematics, University of Minnesota)
  3. Matt Jacobs (Postdoc, UCLA)
  4. Mahmood Ettehad (Postdoc, IMA)
  5. Jason Setiadi
  6. Kevin Miller (Postdoc, Oden Institute)

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

graphlearning-1.5.1.tar.gz (77.3 kB view hashes)

Uploaded Source

Built Distributions

graphlearning-1.5.1-cp311-cp311-win_amd64.whl (87.1 kB view hashes)

Uploaded CPython 3.11 Windows x86-64

graphlearning-1.5.1-cp311-cp311-macosx_10_9_universal2.whl (131.6 kB view hashes)

Uploaded CPython 3.11 macOS 10.9+ universal2 (ARM64, x86-64)

graphlearning-1.5.1-cp310-cp310-win_amd64.whl (87.1 kB view hashes)

Uploaded CPython 3.10 Windows x86-64

graphlearning-1.5.1-cp310-cp310-macosx_11_0_x86_64.whl (99.9 kB view hashes)

Uploaded CPython 3.10 macOS 11.0+ x86-64

graphlearning-1.5.1-cp39-cp39-win_amd64.whl (87.1 kB view hashes)

Uploaded CPython 3.9 Windows x86-64

graphlearning-1.5.1-cp39-cp39-macosx_11_0_x86_64.whl (99.9 kB view hashes)

Uploaded CPython 3.9 macOS 11.0+ x86-64

graphlearning-1.5.1-cp38-cp38-win_amd64.whl (87.1 kB view hashes)

Uploaded CPython 3.8 Windows x86-64

graphlearning-1.5.1-cp38-cp38-macosx_11_0_x86_64.whl (99.9 kB view hashes)

Uploaded CPython 3.8 macOS 11.0+ x86-64

graphlearning-1.5.1-cp37-cp37m-win_amd64.whl (87.0 kB view hashes)

Uploaded CPython 3.7m Windows x86-64

graphlearning-1.5.1-cp37-cp37m-macosx_11_0_x86_64.whl (99.7 kB view hashes)

Uploaded CPython 3.7m macOS 11.0+ x86-64

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page