tensorflow / ranking

Learning to Rank in TensorFlow
Apache License 2.0
2.74k stars 474 forks source link

Inputs are not scaled according to the temperature in pair wise losses #354

Open nicolaspi opened 1 year ago

nicolaspi commented 1 year ago

The class class _PairwiseLoss(_RankingLoss): overrides def call(...) from its parent without applying the temperature scaling (unlike its parent):

https://github.com/tensorflow/ranking/blob/c46cede726fd453e0aaa6097871d23dc8e465bdc/tensorflow_ranking/python/keras/losses.py#L322