allegro / allRank

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

About the input of the loss function #40

Open bugczw opened 3 years ago

bugczw commented 3 years ago

In the loss function 'neuralNDCG', the input 'y_pred' is the prediction from the model, and it's shape is '[batch_size, slate_length]'. What does the 'slate_length' mean? Is each element in 'y_pred' a similarity score or a ranking value? Can you show a example of the loss function inputs?

HanielF commented 2 years ago

refer to https://github.com/allegro/allRank/issues/29