metric-learn: Metric Learning Algorithms in Python

William de Vazelhes (1) , Cj Carey (2) , Yuan Tang (3) , Nathalie Vauquier (1) , Aurélien Bellet (1)

1 MAGNET - Machine Learning in Information Networks
2 Google Inc.
3 Ant Financial

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.

Mots clés: Machine Learning; Python; Metric Learning; Scikit-learn

Domaines: Apprentissage [cs.LG]; Machine Learning [stat.ML]; Génie logiciel [cs.SE]

William de Vazelhes, Cj Carey, Yuan Tang, Nathalie Vauquier, Aurélien Bellet. metric-learn: Metric Learning Algorithms in Python. 2019. ⟨hal-02376986⟩