Skip to main content

Python package for unsupervised and supervised self-organizing maps (SOM)

Project description

PyPi - Code Version PyPI - Python Version PyPI - License Travis.CI Status Documentation Status Codecov Codacy Badge

SUSI: SUpervised Self-organIzing maps in Python

Python package for unsupervised and supervised self-organizing maps (SOM)


We present the SUSI package for Python. It includes a fully functional SOM for unsupervised and supervised tasks. The class structure is set up as follows:

  • SOMClustering: Unsupervised SOM for clustering

    • SOMEstimator: Base class for supervised SOMs

      • SOMRegressor: Regression SOM

      • SOMClassifier: Classification SOM


3-Clause BSD license


Felix M. Riese


see Citation and in the bibtex file


read the docs




With PyPi:

pip3 install susi


git clone
cd susi/
python install


Python 3 with:

  • joblib

  • numpy

  • scikit-learn

  • scipy


Regression in python3:

import susi

som = susi.SOMRegressor(), y_train)
print(som.score(X_test, y_test))

Classification in python3:

import susi

som = susi.SOMClassifier(), y_train)
print(som.score(X_test, y_test))

Code examples as Jupyter Notebooks:


The bibtex file including both references is available here.


Felix M. Riese and S. Keller, “SUSI: Supervised Self-Organizing Maps for Regression and Classification in Python”, arXiv:1903.11114, 2019. Submitted to an ISPRS conference.

    author = {Riese, Felix~M. and Keller, Sina},
    title = {SUSI: Supervised Self-Organizing Maps for Regression and Classification in Python},
    year = {2019},
    notes = {Submitted to an ISPRS conference},
        archivePrefix = {arXiv},
        eprint = {1903.11114},
        primaryClass = {cs.LG},
        url = {}


Felix M. Riese, “SUSI: SUpervised Self-organIzing maps in Python”, 10.5281/zenodo.2609130, 2019.
    author = {Riese, Felix~M.},
    title = {{SUSI: SUpervised Self-organIzing maps in Python}},
    year = {2019},
    DOI = {10.5281/zenodo.2609130},
    publisher = {Zenodo},
    howpublished = {\href{}{}}

Change Log

[1.0.3] - 2019-04-09

  • [ADDED] Link to arXiv paper

  • [ADDED] Mexican-hat neighborhood distance weight

  • [ADDED] Possibility for different initialization modes

  • [CHANGED] Simplified initialization of estimators

  • [FIXED] URLs and styles in documentation

  • [FIXED] Colormap in Salinas example

[1.0.2] - 2019-03-27

  • [CHANGED] Moved decreasing_rate() out of SOM classes

  • [FIXED] Removed duplicate constructor for SOMRegressor, fixed fit() params

  • [ADDED] Codecov, Codacy

[1.0.1] - 2019-03-26

  • [ADDED] Config file for Travis

  • [ADDED] Requirements for read-the-docs documentation

[1.0.0] - 2019-03-26

  • Initial release

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 (15.7 kB view hashes)

Uploaded source

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Huawei Huawei PSF Sponsor Microsoft Microsoft PSF Sponsor NVIDIA NVIDIA PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page