search

allegro / allRank

allRank is a framework for training learning-to-rank neural models based on PyTorch.
Apache License 2.0
870 stars 119 forks source link

code for NeuralNDCG #22

Closed zzchust closed 3 years ago

zzchust commented 3 years ago

Recently, I read a paper about image retrieval, named SmoothAP aimed at optimizing neural network directly through AP metrics. It motives me to take this idea to other metrics, such as NDCG.

As you know， I have seen your paper NeuralNDCG https://docs.mlinpl.org/virtual-event/2020/posters/37-NeuralNDCG_Direct_Optimisation_of_a_Ranking_Metric_via_Differentiable_Relaxation_of_Sorting.pdf. I am so excited about this work. It fully implemented what I am thinking about.

I cant waited to use it in my task, would you mind to share this code to me !

sincere thanks！

sadaharu-inugami commented 3 years ago

Thank you for your kind words. We are currently discussing the code release so it should be available in the near future (let's say, a few weeks).

sadaharu-inugami commented 3 years ago

NeuralNDCG has been released in 1.4 version.