NVIDIA-Merlin / Transformers4Rec

Transformers4Rec is a flexible and efficient library for sequential and session-based recommendation and works with PyTorch.
https://nvidia-merlin.github.io/Transformers4Rec/main
Apache License 2.0
1.08k stars 142 forks source link

Fixes on RecallAt metric for cut-off 1 and for scalar cut-off #720

Closed gabrielspmoreira closed 1 year ago

gabrielspmoreira commented 1 year ago

Fixes #464, Fixes #700

Goals :soccer:

Testing Details :mag:

Added new unit tests to cover RecallAt metric, for both 2D and 3D inputs, with cut-off 1 and with scalar/unique cut-off. I have also used the paper reproducibility script to test it and now recall@1 is smaller than the upper cut-offs

github-actions[bot] commented 1 year ago

Documentation preview

https://nvidia-merlin.github.io/Transformers4Rec/review/pr-720