metric-learn: Metric Learning Algorithms in Python

TL;DR

metric-learn is a Python package that offers a unified interface for various metric learning algorithms, enabling easy integration with scikit-learn for model selection and pipeline construction.

Contribution

It introduces an open source, scikit-learn-compatible Python library for supervised and weakly-supervised metric learning algorithms.

Findings

Provides a unified, easy-to-use interface for metric learning algorithms.

Enables seamless integration with scikit-learn workflows.

Thoroughly tested and available on PyPi.

Abstract

metric-learn is an open source Python package implementing supervised and weakly-supervised distance metric learning algorithms. As part of scikit-learn-contrib, it provides a unified interface compatible with scikit-learn which allows to easily perform cross-validation, model selection, and pipelining with other machine learning estimators. metric-learn is thoroughly tested and available on PyPi under the MIT licence.