Transformers4Rec icon indicating copy to clipboard operation
Transformers4Rec copied to clipboard

[QST] Cross-entropy and pairwise losses are supported in Next Item Prediction

Open ralgond opened this issue 2 years ago • 0 comments

❓ Cross-entropy and pairwise losses are supported in Next Item Prediction

I found the text as following on page https://nvidia-merlin.github.io/Transformers4Rec/stable/model_definition.html?highlight=pairwise%20loss

Next Item Prediction: Predicts next items for a given sequence of interactions. During training, the prediction can be the next item or randomly selected items depending on the masking scheme. For inference, the intended purpose is to always predict the next interacted item. Cross-entropy and pairwise losses are supported.

Could you please tell me how to use pairwise losses instead of Cross-entorpy please?

ralgond avatar Oct 16 '23 06:10 ralgond