search

KevinMusgrave / pytorch-metric-learning

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
https://kevinmusgrave.github.io/pytorch-metric-learning/
MIT License
6.02k stars 658 forks source link

NDCGLoss #675

Open celsofranssa opened 12 months ago

celsofranssa commented 12 months ago

Since metric learning is very useful for ranking tasks, I'd like to suggest including the ApproxNDCGLoss and other ranking losses such as L2R losses using the PML approach.