PyPI Stats
  • Insights
  • PyPI
  • GitHub
  • Search
  • Compare
  • Advisories
  • Ecosystem
  • About
Home

Metric Learning Python Packages

Python packages with the GitHub topic metric-learning. Sorted by relevance, with stars and monthly downloads.
KevinMusgrave
pytorch-metric-learning

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

3.7M 6K 660
catalyst-team
catalyst

Accelerated deep learning R&D

27K 3K 400
tensorflow
tfsim-nightly

TensorFlow Similarity is a python package focused on making similarity learning quick and easy.

7K 1K 105
scikit-learn-contrib
metric-learn

Metric learning algorithms in Python

7K 1K 232
OML-Team
open-metric-learning

OML is a PyTorch-based framework to train and validate the models producing high-quality embeddings.

2K 989 77
jina-ai
finetuner

Task-oriented finetuning for better embeddings on neural search.

2K 2K 68
tensorflow
tensorflow-similarity

TensorFlow Similarity is a python package focused on making similarity learning quick and easy.

2K 1K 105
johny-c
pylmnn

Large Margin Nearest Neighbors implementation in python

1K 49 13
qdrant
quaterion

Blazing fast framework for fine-tuning similarity learning models

1K 660 47
nmichlo
disent

🧶 Modular VAE disentanglement framework for python built with PyTorch Lightning ▸ Including metrics and datasets ▸ With strongly supervised, weakly supervised and unsupervised methods ▸ Easily configured and run with Hydra config ▸ Inspired by disentanglement_lib

1K 134 18
KevinMusgrave
powerful-benchmarker

A library for ML benchmarking. It's powerful.

1K 439 42
romue404
metric-learning-layers

A simple PyTorch package that includes the most common metric learning layers

1K 1 0
MPI-Dortmund
tomotwin-cryoet

Picking procedure for cryo em tomography

880 43 10
DaminK
ggml-ot

Supervised OT with Global Ground Metric Learning

743 4 0
bm2-lab
scmdml

scmdml

545 4 4
zbr17
gedml

GeDML is an easy-to-use generalized deep metric learning library, which contains state-of-the-art deep metric learning algorithms and auxiliary modules to build end-to-end compute vision systems

540 36 3
raamana
kernelmethods

Kernel methods and classes for composable kernel learning workflows.

467 43 11
hand10ryo
pytorchcml

Collaborative Metric Learning implemented by PyTorch

463 20 3
patrickfrank1
chesspos

Embedding based chess position search and embedding learning for chess positions

439 16 6
Ximilar-com
tf-metric-learning

Minimalistic TensorFlow2+ deep metric/similarity learning library with loss functions, miners, and utils as embedding projector.

382 38 5
gjhuizing
wsingular

Python package for the ICML 2022 paper "Unsupervised Ground Metric Learning Using Wasserstein Singular Vectors".

259 10 1
bm2-lab
mtsc

mtSC

126 4 4
jina-ai
finetuner-client

Finetuner allows one to tune the weights of any deep neural network for better embeddings on search tasks.

102 2K 68
mmkuznecov
ssrlib

A modular Python framework for Self-Supervised Learning with automatic component discovery and intelligent caching

97 4 0
    • Data from PyPI, GitHub, ClickHouse, and BigQuery