Skip to main content

Python implementations of metric learning algorithms

Project description

Travis-CI Build Status License PyPI version Code coverage


Metric Learning algorithms in Python.


  • Large Margin Nearest Neighbor (LMNN)

  • Information Theoretic Metric Learning (ITML)

  • Sparse Determinant Metric Learning (SDML)

  • Least Squares Metric Learning (LSML)

  • Neighborhood Components Analysis (NCA)

  • Local Fisher Discriminant Analysis (LFDA)

  • Relative Components Analysis (RCA)

  • Metric Learning for Kernel Regression (MLKR)

  • Mahalanobis Metric for Clustering (MMC)


  • Python 2.7+, 3.4+

  • numpy, scipy, scikit-learn>=0.20.3

Optional dependencies

  • For SDML, using skggm will allow the algorithm to solve problematic cases (install from commit a0ed406).

  • For running the examples only: matplotlib


Run pip install metric-learn to download and install from PyPI.

Run python install for default installation.

Run pytest test to run all tests (you will need to have the pytest package installed).


See the sphinx documentation for full documentation about installation, API, usage, and examples.

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

metric-learn-0.5.0.tar.gz (71.3 kB view hashes)

Uploaded source

Built Distribution

metric_learn-0.5.0-py2.py3-none-any.whl (60.9 kB view hashes)

Uploaded py2 py3

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