Transformers4Rec icon indicating copy to clipboard operation
Transformers4Rec copied to clipboard
verified publisher icon NVIDIA-Merlin

Improve performance of ranking metrics calculation

Open sararb opened this issue 4 years ago • 1 comments

  • The current implementation of ranking metrics computes the top k scores within each metric class using the method extract_topk which is computationally expensive.
  • A possible solution is to Include the extraction of the top-k scores within the NextItemPredictionTask class and provide the ranked scores to the ranking metrics.

sararb avatar Sep 01 '21 17:09 sararb

@sararb I would like to take this issue. Is that ok?

murphp15 avatar Dec 20 '21 19:12 murphp15